| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | tensor_util.py | 133 def _shape_tensor_compatible(expected_shape, actual_shape): argument
169 def _is_shape(expected_shape, actual_tensor, actual_shape=None): argument
189 def _assert_shape_op(expected_shape, actual_tensor): argument
232 def with_shape(expected_shape, tensor): argument
|
| D | tensor_util_test.py | 83 def _assert_with_shape(self, tensor, expected_value, expected_shape, argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_expand_dims_op_test.py | 116 expected_shape=None): argument
|
| D | ragged_to_tensor_op_test.py | 102 expected_shape=None): argument
|
| D | ragged_constant_value_op_test.py | 148 expected_shape=None, argument
|
| D | ragged_stack_op_test.py | 325 expected_shape=None): argument
|
| D | ragged_const_op_test.py | 144 expected_shape=None, argument
|
| D | ragged_concat_op_test.py | 236 expected_shape=None): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_add_op_test.cc | 89 Tensor expected_shape(allocator(), DT_INT64, in TEST_F() local
|
| D | sparse_reduce_sum_op_test.cc | 118 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
|
| D | padding_fifo_queue.cc | 256 const PartialTensorShape expected_shape = in ValidateManyTuple() local
|
| D | queue_base.cc | 167 const TensorShape expected_shape = ManyOutShape(i, batch_size); in ValidateManyTuple() local
|
| D | lookup_table_op.cc | 389 TensorShape expected_shape({num_elements}); in Find() local
456 TensorShape expected_shape({batch_size}); in Insert() local
481 TensorShape expected_shape({key.dim_size(0)}); in Remove() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.cc | 73 const auto& expected_shape = computation->parameter_instruction(i)->shape(); in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | edit_distance_op_test.py | 45 expected_shape, argument
|
| D | split_op_test.py | 248 def _testEmpty(self, x, dim, num, expected_shape): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 361 Shape expected_shape = in TEST_F() local
399 Shape expected_shape = ShapeUtil::MakeShapeWithLayout(F32, {3, 2}, {1, 0}); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_block_diag_test.py | 35 def _block_diag_dense(expected_shape, blocks): argument
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 38 TensorShape expected_shape, error::Code expected_code) { in TestBitcastOp()
|
| /external/tensorflow/tensorflow/core/example/ |
| D | example_parser_configuration_test.cc | 212 TensorShape expected_shape({1}); in TEST_F() local
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | const_op_test.cc | 39 TensorShape expected_shape) { in ExpectTypeAndShape()
|
| /external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/ |
| D | sequence_feature_column_test.py | 495 self, sparse_input_args, expected_shape): argument
530 self, sparse_input_args, expected_shape): argument
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | sequence_feature_column_test.py | 488 self, sparse_input_args, expected_shape): argument
523 self, sparse_input_args, expected_shape): argument
|