Searched refs:ForEachSubshapeWithStatus (Results 1 – 17 of 17) sorted by relevance
21 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in Preprocess()
2523 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in HandleWhile()
980 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in GetOutputInfo()
72 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in TransferLiteralFromDevice()112 return ShapeUtil::ForEachSubshapeWithStatus( in TransferLiteralToDeviceAsync()
66 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in Build()
375 return ShapeUtil::ForEachSubshapeWithStatus( in SetInstructionLayout()413 return ShapeUtil::ForEachSubshapeWithStatus( in SetInstructionLayout()929 return ShapeUtil::ForEachSubshapeWithStatus( in CheckParameterLayout()1643 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in PropagateOperandConstraint()1679 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in PropagateOperandConstraint()2370 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in PropagateComputationLayouts()
291 return ShapeUtil::ForEachSubshapeWithStatus( in WriteTupleIndexTablesAsync()
504 Status status = ShapeUtil::ForEachSubshapeWithStatus( in HaveDisjointSlices()
1630 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckMixedPrecisionOperands()
33 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in MakeArray()
145 return xla::ShapeUtil::ForEachSubshapeWithStatus( in FixTupleTableAsync()212 TF_RETURN_IF_ERROR(xla::ShapeUtil::ForEachSubshapeWithStatus( in UpdateDynamicInputs()
252 return xla::ShapeUtil::ForEachSubshapeWithStatus(parameter_shape, in InputShapeMatches()
484 static Status ForEachSubshapeWithStatus(const Shape& shape,
1124 /* static */ Status ShapeUtil::ForEachSubshapeWithStatus( in ForEachSubshapeWithStatus() function in xla::ShapeUtil
216 TF_RETURN_IF_ERROR(xla::ShapeUtil::ForEachSubshapeWithStatus( in UpdateDynamicInputs()
1604 TF_RETURN_IF_ERROR(xla::ShapeUtil::ForEachSubshapeWithStatus( in Initialize()