| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | linalg_ops.cc | 37 ShapeHandle batch_shape; in MakeBatchSquareMatrix() local
166 ShapeHandle batch_shape; in SelfAdjointEigV2ShapeFn() local
193 ShapeHandle batch_shape; in LuShapeFn() local
219 ShapeHandle batch_shape; in QrShapeFn() local
250 ShapeHandle batch_shape; in SvdShapeFn() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | linalg_ops_common.cc | 96 TensorShape batch_shape; in Compute() local
122 TensorShapes* input_matrix_shapes, TensorShape* batch_shape) { in AnalyzeInputs()
170 const TensorShape& batch_shape, TensorOutputs* outputs, in PrepareOutputs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/ |
| H A D | slicing.py | 40 def _broadcast_parameter_with_batch_shape( argument
106 param, param_ndims_to_matrix_ndims, slices, batch_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/proto/ |
| H A D | decode_proto_op_test_base.py | 70 def _compareProtos(self, batch_shape, sizes, fields, field_dict): argument
145 def _runDecodeProtoTests(self, fields, case_sizes, batch_shape, batch, argument
|
| /aosp_15_r20/external/pytorch/torch/distributions/ |
| H A D | relaxed_categorical.py | 50 def expand(self, batch_shape, _instance=None): argument
128 def expand(self, batch_shape, _instance=None): argument
|
| H A D | relaxed_bernoulli.py | 63 def expand(self, batch_shape, _instance=None): argument
138 def expand(self, batch_shape, _instance=None): argument
|
| H A D | chi2.py | 29 def expand(self, batch_shape, _instance=None): argument
|
| H A D | logistic_normal.py | 46 def expand(self, batch_shape, _instance=None): argument
|
| H A D | lkj_cholesky.py | 85 def expand(self, batch_shape, _instance=None): argument
|
| H A D | weibull.py | 47 def expand(self, batch_shape, _instance=None): argument
|
| H A D | pareto.py | 35 def expand(self, batch_shape, _instance=None): argument
|
| H A D | kumaraswamy.py | 64 def expand(self, batch_shape, _instance=None): argument
|
| H A D | fishersnedecor.py | 46 def expand(self, batch_shape, _instance=None): argument
|
| H A D | gumbel.py | 53 def expand(self, batch_shape, _instance=None): argument
|
| H A D | log_normal.py | 38 def expand(self, batch_shape, _instance=None): argument
|
| H A D | inverse_gamma.py | 47 def expand(self, batch_shape, _instance=None): argument
|
| H A D | poisson.py | 55 def expand(self, batch_shape, _instance=None): argument
|
| H A D | studentT.py | 68 def expand(self, batch_shape, _instance=None): argument
|
| H A D | geometric.py | 75 def expand(self, batch_shape, _instance=None): argument
|
| H A D | half_cauchy.py | 40 def expand(self, batch_shape, _instance=None): argument
|
| H A D | half_normal.py | 40 def expand(self, batch_shape, _instance=None): argument
|
| /aosp_15_r20/external/pytorch/test/distributions/ |
| H A D | test_transforms.py | 326 def test_transformed_distribution_shapes(batch_shape, event_shape, dist): argument
481 def test_compose_reshape(batch_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A D | linalg_ops_test.py | 207 def test_eye_no_placeholder(self, num_rows, num_columns, batch_shape, dtype): argument
238 self, num_rows, num_columns, batch_shape, dtype): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | linalg_ops_impl.py | 31 batch_shape=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | matrix_band_part_test.py | 165 def testMatrixBandPart(self, batch_shape, rows, cols): argument
|