| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | slicing.py | 40 def _broadcast_parameter_with_batch_shape( argument
106 param, param_ndims_to_matrix_ndims, slices, batch_shape): argument
|
| D | linear_operator_zeros.py | 126 batch_shape=None, argument
|
| D | linear_operator_identity.py | 206 batch_shape=None, argument
|
| D | linear_operator.py | 346 def batch_shape(self): member in LinearOperator
|
| /external/tensorflow/tensorflow/core/ops/ |
| 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
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| 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()
|
| D | lu_op.cc | 80 TensorShape batch_shape; in Compute() local
|
| D | lu_op_gpu.cu.cc | 103 TensorShape batch_shape; in ComputeAsync() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/proto/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_band_part_test.py | 165 def testMatrixBandPart(self, batch_shape, rows, cols): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops_impl.py | 31 batch_shape=None, argument
|
| D | linalg_ops.py | 196 batch_shape=None, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | bernoulli_test.py | 44 def make_bernoulli(batch_shape, dtype=dtypes.int32): argument
|
| D | categorical_test.py | 36 def make_categorical(batch_shape, num_classes, dtype=dtypes.int32): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| 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
|
| D | matrix_solve_ls_op_test.py | 86 batch_shape=()): argument
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | transformed_distribution.py | 238 batch_shape=None, argument
|
| D | distribution.py | 653 def batch_shape(self): member in Distribution
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op.cc | 450 TensorShape batch_shape = batch->shape(); in CallFunction() local
|