| /external/XNNPACK/test/ |
| D | transpose-normalization-tester.h | 69 inline TransposeNormalizationTester& expected_shape(const std::vector<size_t> expected_shape) { in expected_shape() argument
74 inline const std::vector<size_t>& expected_shape() const { return this->expected_shape_; } in expected_shape() function
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_expand_dims_op_test.py | 111 expected_shape=None): argument
|
| D | ragged_stack_op_test.py | 346 expected_shape=None): argument
|
| D | ragged_constant_value_op_test.py | 188 expected_shape=None, argument
|
| D | ragged_getitem_test.py | 121 def _TestGetItem(self, rt, slice_spec, expected, expected_shape=None): argument
553 def testWithUniformRowLength(self, slice_spec, expected, expected_shape): argument
|
| D | ragged_matmul_op_test.py | 193 def testMatmul(self, a, b, expected_shape=None, **kwargs): argument
|
| D | ragged_to_tensor_op_test.py | 326 expected_shape=None): argument
481 self, input_shape, to_tensor_shape, expected_shape): argument
|
| D | ragged_concat_op_test.py | 231 expected_shape=None): argument
|
| D | ragged_const_op_test.py | 179 expected_shape=None, argument
|
| D | ragged_from_tensor_op_test.py | 354 expected_shape=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.cc | 85 const auto& expected_shape = computation->parameter_instruction(i)->shape(); in ExecuteAsyncOnStream() local
117 const auto& expected_shape = computation->parameter_instruction(p)->shape(); in ExecuteAsyncOnStream() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | bincount_ops_test.py | 172 expected_shape, argument
379 expected_shape, argument
529 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_add_op_test.cc | 42 float y_max_value, const std::vector<int64_t>& expected_shape, in TestAdd()
129 std::vector<int64_t> expected_shape; in TestAddShape() local
|
| D | quantized_mul_op_test.cc | 42 float y_max_value, const std::vector<int64_t>& expected_shape, in TestMul()
129 std::vector<int64_t> expected_shape; in TestMulShape() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | edit_distance_op_test.py | 41 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 | 35 def _block_lower_triangular_dense(expected_shape, blocks): argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | spmd_expander.cc | 134 const auto expected_shape = std::get<1>(expanded_and_true_global_shape); in ExpandOpAndSetLayout() local
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor_test.py | 301 def testFromFieldsAndRank(self, fields, rank, expected_shape): argument
555 expected_shape=None, 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/compiler/xla/service/gpu/ |
| D | matmul_utils_test.cc | 44 absl::string_view expected_shape; member
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 37 TensorShape expected_shape, error::Code expected_code) { in TestBitcastOp()
|