| /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 | 93 TensorShape batch_shape; in Compute() local
119 TensorShapes* input_matrix_shapes, TensorShape* batch_shape) { in AnalyzeInputs()
163 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/ops/ |
| D | linalg_ops_impl.py | 35 batch_shape=None, argument
|
| D | linalg_ops.py | 200 batch_shape=None, argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_band_part_test.py | 169 def testMatrixBandPart(self, batch_shape, rows, cols): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/proto/ |
| D | decode_proto_op_test_base.py | 74 def _compareProtos(self, batch_shape, sizes, fields, field_dict): argument
149 def _runDecodeProtoTests(self, fields, case_sizes, batch_shape, batch, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_band_part_op.cc | 57 TensorShape batch_shape = input_shape; in Compile() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | linalg_ops_test.py | 217 def test_eye_no_placeholder(self, num_rows, num_columns, batch_shape, dtype): argument
248 self, num_rows, num_columns, batch_shape, dtype): argument
|
| D | matrix_solve_ls_op_test.py | 89 batch_shape=()): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | bernoulli_test.py | 48 def make_bernoulli(batch_shape, dtype=dtypes.int32): argument
|
| D | categorical_test.py | 40 def make_categorical(batch_shape, num_classes, dtype=dtypes.int32): argument
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_zeros.py | 129 batch_shape=None, argument
|
| D | linear_operator_identity.py | 209 batch_shape=None, argument
|
| D | linear_operator.py | 342 def batch_shape(self): member in LinearOperator
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | transformed_distribution.py | 242 batch_shape=None, argument
|
| D | distribution.py | 660 def batch_shape(self): member in Distribution
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op.cc | 440 TensorShape batch_shape = batch->shape(); in CallFunction() local
|