Searched refs:MinorToMajorOnlyInLayout (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_layout.cc | 66 equal.MinorToMajorOnlyInLayout(); in MatchesLayoutInShape()
72 equal.MinorToMajorOnlyInLayout(); in MatchesLayoutInShape()
77 equal.MinorToMajorOnlyInLayout(); in MatchesLayoutInShape()
|
| D | shape.h | 211 Equal& MinorToMajorOnlyInLayout() { in MinorToMajorOnlyInLayout() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.h | 152 equal.MinorToMajorOnlyInLayout();
164 equal.MinorToMajorOnlyInLayout();
|
| D | hlo_matchers.cc | 176 compare.MinorToMajorOnlyInLayout(); in MatchAndExplain()
|
| D | layout_assignment.cc | 861 if (!Shape::Equal().MinorToMajorOnlyInLayout().IgnoreDynamicDimension()( in CheckParameterLayout()
906 if (Shape::Equal().MinorToMajorOnlyInLayout()(target_shape, in CreateCopyWithNewLayout()
953 if (Shape::Equal().MinorToMajorOnlyInLayout()(operand_layout.shape(), in CopyOperandIfLayoutsDiffer()
1042 .MinorToMajorOnlyInLayout()(instruction_subshape, in CheckLayouts()
1106 Shape::Equal().IgnoreDynamicDimension().MinorToMajorOnlyInLayout()( in CheckLayouts()
2216 Shape::Equal().IgnoreDynamicDimension().MinorToMajorOnlyInLayout()( in PropagateComputationLayouts()
|
| D | transfer_manager.cc | 184 if (!Shape::Equal().MinorToMajorOnlyInLayout()(HostShapeToDeviceShape(shape), in TransferArrayFromDevice()
|
| D | bfloat16_propagation.cc | 790 if (!Shape::Equal().MinorToMajorOnlyInLayout()(hlo->literal().shape(), in ResolveConvertedConstants()
|
| D | hlo_evaluator.cc | 249 if (!Shape::Equal().MinorToMajorOnlyInLayout()(computation_shape, in Evaluate()
495 DCHECK(Shape::Equal().MinorToMajorOnlyInLayout()(parameter->shape(), in HandleParameter()
|
| D | hlo_instructions.cc | 1415 CHECK(Shape::Equal().MinorToMajorOnlyInLayout()(literal_->shape(), in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.cc | 331 TF_RET_CHECK(Shape::Equal().MinorToMajorOnlyInLayout()( in CreateFromProto()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.cc | 93 } else if (!Shape::Equal().MinorToMajorOnlyInLayout()(expected_shape, in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 98 if (!Shape::Equal().MinorToMajorOnlyInLayout()(shape, device_shape)) { in AllocateOutputMemoryWithInputReuse()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 283 if (xla::Shape::Equal().MinorToMajorOnlyInLayout()(shape, device_shape)) { in PopulateInputs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3729 TF_RET_CHECK(Shape::Equal().MinorToMajorOnlyInLayout()( in Run()
3736 TF_RET_CHECK(Shape::Equal().MinorToMajorOnlyInLayout()( in Run()
|