| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.cc | 85 const auto& expected_shape = computation->parameter_instruction(i)->shape(); in ExecuteAsyncOnStream() local
115 const auto& expected_shape = computation->parameter_instruction(p)->shape(); in ExecuteAsyncOnStream() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_reduce_op.cc | 89 xla::Shape expected_shape; in Compile() local
210 xla::Shape expected_shape = in Compile() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_expand_dims_op_test.py | 115 expected_shape=None): argument
|
| D | ragged_stack_op_test.py | 350 expected_shape=None): argument
|
| D | ragged_constant_value_op_test.py | 192 expected_shape=None, argument
|
| D | ragged_getitem_test.py | 125 def _TestGetItem(self, rt, slice_spec, expected, expected_shape=None): argument
557 def testWithUniformRowLength(self, slice_spec, expected, expected_shape): argument
|
| D | ragged_to_tensor_op_test.py | 331 expected_shape=None): argument
486 self, input_shape, to_tensor_shape, expected_shape): argument
|
| D | ragged_matmul_op_test.py | 174 def testMatmul(self, a, b, expected_shape=None, **kwargs): argument
|
| D | ragged_concat_op_test.py | 235 expected_shape=None): argument
|
| D | ragged_const_op_test.py | 183 expected_shape=None, argument
|
| D | ragged_from_tensor_op_test.py | 358 expected_shape=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 358 Shape expected_shape = in TEST_F() local
395 Shape expected_shape = ShapeUtil::MakeShapeWithLayout(F32, {3, 2}, {1, 0}); in TEST_F() local
421 Shape expected_shape = ShapeUtil::MakeShapeWithLayout(C64, {8, 32}, {1, 0}); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | bincount_ops_test.py | 153 expected_shape, argument
349 expected_shape, argument
496 expected_shape, argument
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | sequence_feature_column_test.py | 490 self, sparse_input_args, expected_shape): argument
524 self, sparse_input_args, expected_shape): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_add_op_test.cc | 90 Tensor expected_shape(allocator(), DT_INT64, in TEST_F() local
|
| D | sparse_reduce_sum_op_test.cc | 119 Tensor expected_shape(allocator(), DT_INT64, {1}); in TEST_F() local
|
| D | quantized_mul_op_test.cc | 42 float y_max_value, const std::vector<int64>& expected_shape, in TestMul()
129 std::vector<int64> expected_shape; in TestMulShape() local
|
| D | quantized_add_op_test.cc | 42 float y_max_value, const std::vector<int64>& expected_shape, in TestAdd()
129 std::vector<int64> expected_shape; in TestAddShape() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | edit_distance_op_test.py | 45 expected_shape, argument
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | signature_flattening_test.cc | 32 const PartialTensorShape& expected_shape, in TensorSpecsAreEqual()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_block_lower_triangular_test.py | 38 def _block_lower_triangular_dense(expected_shape, blocks): argument
|
| D | linear_operator_block_diag_test.py | 39 def _block_diag_dense(expected_shape, blocks): argument
|
| /external/tensorflow/tensorflow/core/example/ |
| D | example_parser_configuration_test.cc | 212 TensorShape expected_shape({1}); in TEST_F() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 107 const absl::optional<std::vector<int64_t>>& expected_shape) { in CheckOutputShapes()
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | const_op_test.cc | 39 TensorShape expected_shape) { in ExpectTypeAndShape()
|