Searched refs:IgnoreLayout (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.h | 189 Equal& IgnoreLayout() { in IgnoreLayout() function
|
| D | shape_util.cc | 655 return Shape::Equal().IgnoreDynamicDimension().IgnoreLayout()(lhs, rhs); in Compatible()
663 .IgnoreLayout()(lhs, rhs); in CompatibleIgnoringElementType()
670 .IgnoreLayout() in CompatibleKind()
680 .IgnoreLayout()(lhs, rhs); in CompatibleIgnoringFpPrecision()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_all_reduce_code_motion.cc | 444 CHECK(Shape::Equal().IgnoreLayout()(old_buffer->shape(), in CreateSinkedAllReduces()
|
| D | while_loop_analysis.cc | 621 CHECK(Shape::Equal().IgnoreLayout()(cond_result_pred.shape(), in ComputeWhileLoopTripCountUpperBound()
|
| D | dynamic_dimension_inference.cc | 1674 CHECK(Shape::Equal().IgnoreLayout()(replace->shape(), in ReplaceAllDynamicDimensionUsesWith()
1676 CHECK(Shape::Equal().IgnoreLayout()(with->shape(), in ReplaceAllDynamicDimensionUsesWith()
|
| D | hlo_computation.cc | 351 if (!Shape::Equal().IgnoreLayout()(new_root_instruction->shape(), in set_root_instruction()
|
| D | hlo_instruction.cc | 1094 CHECK(Shape::Equal().IgnoreLayout()(shape, ShapeUtil::MakeShape(U32, {}))) in CreateReplicaId()
1102 CHECK(Shape::Equal().IgnoreLayout()(shape, ShapeUtil::MakeShape(U32, {}))) in CreatePartitionId()
|
| D | shape_inference.cc | 3144 .IgnoreLayout() in InferSelectShape()
|