Searched refs:MinorToMajorOnlyInLayout (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_layout.cc | 53 equal.MinorToMajorOnlyInLayout(); in MatchesLayoutInShape()
|
| D | shape.h | 205 Equal& MinorToMajorOnlyInLayout() { in MinorToMajorOnlyInLayout() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.h | 147 equal.MinorToMajorOnlyInLayout();
159 equal.MinorToMajorOnlyInLayout();
|
| D | layout_assignment.cc | 774 if (!Shape::Equal().MinorToMajorOnlyInLayout().IgnoreDynamicDimension()( in CheckParameterLayout()
819 if (Shape::Equal().MinorToMajorOnlyInLayout()(target_shape, in CreateCopyWithNewLayout()
866 if (Shape::Equal().MinorToMajorOnlyInLayout()(operand_layout.shape(), in CopyOperandIfLayoutsDiffer()
954 .MinorToMajorOnlyInLayout()(instruction_subshape, in CheckLayouts()
1018 Shape::Equal().IgnoreDynamicDimension().MinorToMajorOnlyInLayout()( in CheckLayouts()
2089 Shape::Equal().IgnoreDynamicDimension().MinorToMajorOnlyInLayout()( in PropagateComputationLayouts()
|
| D | transfer_manager.cc | 183 if (!Shape::Equal().MinorToMajorOnlyInLayout()(HostShapeToDeviceShape(shape), in TransferArrayFromDevice()
|
| D | bfloat16_propagation.cc | 795 if (!Shape::Equal().MinorToMajorOnlyInLayout()(hlo->literal().shape(), in ResolveConvertedConstants()
|
| D | hlo_evaluator.cc | 248 if (!Shape::Equal().MinorToMajorOnlyInLayout()(computation_shape, in Evaluate()
460 DCHECK(Shape::Equal().MinorToMajorOnlyInLayout()(parameter->shape(), in HandleParameter()
|
| D | hlo_instructions.cc | 1315 CHECK(Shape::Equal().MinorToMajorOnlyInLayout()(literal_->shape(), in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.cc | 330 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 | 86 if (!Shape::Equal().MinorToMajorOnlyInLayout()(host_shape, device_shape)) { in AllocateOutputMemoryWithInputReuse()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 270 if (xla::Shape::Equal().MinorToMajorOnlyInLayout()(shape, device_shape)) { in PopulateInputs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3688 TF_RET_CHECK(Shape::Equal().MinorToMajorOnlyInLayout()( in Run()
3695 TF_RET_CHECK(Shape::Equal().MinorToMajorOnlyInLayout()( in Run()
|