| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 37 ShapeHandle batch_shape; in MakeBatchSquareMatrix() local
95 ShapeHandle batch_shape; in SelfAdjointEigV2ShapeFn() local
122 ShapeHandle batch_shape; in LuShapeFn() local
148 ShapeHandle batch_shape; in QrShapeFn() local
179 ShapeHandle batch_shape; in SvdShapeFn() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_band_part_test.py | 169 def testMatrixBandPart(self, batch_shape, rows, cols): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops_impl.py | 35 batch_shape=None, argument
|
| D | linalg_ops.py | 130 batch_shape=None, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | linalg_ops_common.cc | 92 TensorShape batch_shape; in Compute() local
117 TensorShape* batch_shape) { in AnalyzeInputs()
161 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 | 101 TensorShape batch_shape; in ComputeAsync() local
|
| /external/tensorflow/tensorflow/contrib/proto/python/kernel_tests/ |
| D | decode_proto_op_test_base.py | 72 def _compareProtos(self, batch_shape, sizes, fields, field_dict): argument
144 def _runDecodeProtoTests(self, fields, case_sizes, batch_shape, batch, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | linalg_ops_test.py | 210 def test_eye_no_placeholder(self, num_rows, num_columns, batch_shape, dtype): argument
241 self, num_rows, num_columns, batch_shape, dtype): argument
|
| D | matrix_solve_ls_op_test.py | 88 batch_shape=()): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_set_diag_op.cc | 46 TensorShape batch_shape = input_shape; in Compile() local
|
| D | matrix_band_part_op.cc | 57 TensorShape batch_shape = input_shape; in Compile() local
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | batch_reshape.py | 86 batch_shape, argument
413 def validate_init_args_statically(distribution, batch_shape): argument
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | relaxed_onehot_categorical_test.py | 32 def make_relaxed_categorical(batch_shape, num_classes, dtype=dtypes.float32): argument
|
| D | mvn_full_covariance_test.py | 120 def _random_mu_and_sigma(self, batch_shape, event_shape): argument
|
| D | distribution_test.py | 134 def __init__(self, batch_shape=None, event_shape=None): argument
|
| D | onehot_categorical_test.py | 33 def make_onehot_categorical(batch_shape, num_classes, dtype=dtypes.int32): argument
|
| D | mixture_test.py | 110 def make_univariate_mixture(batch_shape, num_components, use_static_graph): argument
125 def make_multivariate_mixture(batch_shape, num_components, event_shape, argument
|
| D | mvn_tril_test.py | 224 def _random_mu_and_sigma(self, batch_shape, event_shape): argument
|
| D | conditional_distribution_test.py | 34 def __init__(self, batch_shape=None, event_shape=None): argument
|
| D | transformed_distribution_test.py | 310 batch_shape=(), argument
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_zeros.py | 129 batch_shape=None, 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/distributions/ |
| D | transformed_distribution.py | 242 batch_shape=None, argument
|