Home
last modified time | relevance | path

Searched refs:ForEachSubshapeWithStatus (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dgeneric_transfer_manager.cc72 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in TransferLiteralFromDevice()
108 return ShapeUtil::ForEachSubshapeWithStatus( in TransferLiteralToDeviceAsync()
Doptimize_input_output_buffer_alias.cc66 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in Build()
Dlayout_assignment.cc312 return ShapeUtil::ForEachSubshapeWithStatus( in SetInstructionLayout()
767 return ShapeUtil::ForEachSubshapeWithStatus( in CheckParameterLayout()
1408 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in PropagateOperandConstraint()
1450 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in PropagateOperandConstraint()
2054 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in PropagateComputationLayouts()
Dtransfer_manager.cc297 return ShapeUtil::ForEachSubshapeWithStatus( in WriteTupleIndexTablesAsync()
Dbuffer_assignment.cc523 Status status = ShapeUtil::ForEachSubshapeWithStatus( in HaveDisjointSlices()
Dhlo_verifier.cc1107 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckMixedPrecisionOperands()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer_test.cc33 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in MakeArray()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc143 return xla::ShapeUtil::ForEachSubshapeWithStatus( in FixTupleTableAsync()
210 TF_RETURN_IF_ERROR(xla::ShapeUtil::ForEachSubshapeWithStatus( in UpdateDynamicInputs()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc250 return xla::ShapeUtil::ForEachSubshapeWithStatus(parameter_shape, in InputShapeMatches()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h585 static Status ForEachSubshapeWithStatus(const Shape& shape,
Dshape_util.cc1045 /* static */ Status ShapeUtil::ForEachSubshapeWithStatus( in ForEachSubshapeWithStatus() function in xla::ShapeUtil
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc598 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in GetOutputInfo()
Dir_emitter_unnested.cc3780 return ShapeUtil::ForEachSubshapeWithStatus( in CheckWhileBuffersShareAllocation()
3809 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckConditionalBuffersShareAllocation()
3820 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckConditionalBuffersShareAllocation()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc216 TF_RETURN_IF_ERROR(xla::ShapeUtil::ForEachSubshapeWithStatus( in UpdateDynamicInputs()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2365 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in HandleWhile()