Searched refs:ForEachSubshapeWithStatus (Results 1 – 11 of 11) sorted by relevance
83 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in TransferLiteralFromDevice()119 return ShapeUtil::ForEachSubshapeWithStatus( in TransferLiteralToDevice()
121 return ShapeUtil::ForEachSubshapeWithStatus( in WriteTupleIndexTables()
403 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckMixedPrecisionOperands()
280 Status status = ShapeUtil::ForEachSubshapeWithStatus( in HaveDisjointSlices()
372 return ShapeUtil::ForEachSubshapeWithStatus( in SetInstructionLayout()
30 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in Run()
2324 return ShapeUtil::ForEachSubshapeWithStatus( in CheckWhileBuffersShareAllocation()2355 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckConditionalBuffersShareAllocation()2366 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckConditionalBuffersShareAllocation()2374 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckConditionalBuffersShareAllocation()
2229 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in HandleWhile()
457 static Status ForEachSubshapeWithStatus(const Shape& shape,
959 /* static */ Status ShapeUtil::ForEachSubshapeWithStatus( in ForEachSubshapeWithStatus() function in xla::ShapeUtil