Home
last modified time | relevance | path

Searched defs:other_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_layout.cc25 Status ShapeLayout::CopyLayoutFromShape(const Shape& other_shape) { in CopyLayoutFromShape()
Dlayout_util_test.cc43 Shape other_shape = in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dbase_binary_ops_test.h163 const TensorShape& other_shape, in TestOneScalar()
195 const TensorShape& other_shape, in TestOneEffectiveScalar()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc219 Shape other_shape = ShapeUtil::MakeShapeWithLayout(F32, {100, 24}, {0, 1}); in TEST_F() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc599 TensorShape other_shape; in CopyInferenceShape() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2275 bool lhs_matching, const Shape& output_base_shape, const Shape& other_shape, in GetNonContractingPartitionGroupedShardingForOtherOperand()