Searched refs:other_shape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_layout.cc | 25 Status ShapeLayout::CopyLayoutFromShape(const Shape& other_shape) { in CopyLayoutFromShape() argument
26 if (!ShapeUtil::Compatible(other_shape, shape_)) { in CopyLayoutFromShape()
28 ShapeUtil::HumanString(other_shape), in CopyLayoutFromShape()
31 shape_ = other_shape; in CopyLayoutFromShape()
|
| D | shape_layout.h | 53 Status CopyLayoutFromShape(const Shape& other_shape);
|
| D | layout_util_test.cc | 41 Shape other_shape = in TEST_F() local
58 Shape other_tuple2 = ShapeUtil::MakeTupleShape({shape, other_shape}); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_binary_ops_test.h | 160 const TensorShape& other_shape, in TestOneScalar() argument
166 CHECK(other_input.size() <= other_shape.num_elements() && in TestOneScalar()
169 test::RepeatInputToMatchShape(other_input, other_shape.num_elements()); in TestOneScalar()
184 other_shape, repeated_other_input, in TestOneScalar()
185 /*expected_shape=*/other_shape, expected_output, in TestOneScalar()
192 const TensorShape& other_shape, in TestOneEffectiveScalar() argument
199 CHECK(other_input.size() <= other_shape.num_elements() && in TestOneEffectiveScalar()
202 test::RepeatInputToMatchShape(other_input, other_shape.num_elements()); in TestOneEffectiveScalar()
216 TensorShape expected_shape = other_shape; in TestOneEffectiveScalar()
221 op_name, effective_scalar_shape, scalar_input_vector, other_shape, in TestOneEffectiveScalar()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 220 Shape other_shape = ShapeUtil::MakeShapeWithLayout(F32, {100, 24}, {0, 1}); in TEST_F() local
222 auto constant_shape = ShapeUtil::MakeTupleShape({other_shape, rhs_shape}); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 599 TensorShape other_shape; in CopyInferenceShape() local
600 TF_RETURN_IF_ERROR(other->Shape(&other_shape)); in CopyInferenceShape()
601 inference_shape_ = other_shape; in CopyInferenceShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1912 bool lhs_matching, const Shape& output_base_shape, const Shape& other_shape, in GetNonContractingPartitionGroupedShardingForOtherOperand() argument
1947 ShapeUtil::ByteSizeOf(other_shape)) <= in GetNonContractingPartitionGroupedShardingForOtherOperand()
|