Searched refs:ForEachSubshapeWithStatus (Results 1 – 13 of 13) sorted by relevance
83 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in TransferLiteralFromDeviceInternal()120 return ShapeUtil::ForEachSubshapeWithStatus( in TransferLiteralToDeviceAsync()
62 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in Build()
243 return ShapeUtil::ForEachSubshapeWithStatus( in WriteTupleIndexTablesAsync()
302 return ShapeUtil::ForEachSubshapeWithStatus( in SetInstructionLayout()1311 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in PropagateOperandConstraint()1349 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in PropagateOperandConstraint()
32 return ShapeUtil::ForEachSubshapeWithStatus( in VerifyNotSparse()786 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckMixedPrecisionOperands()
461 Status status = ShapeUtil::ForEachSubshapeWithStatus( in HaveDisjointSlices()
29 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in Run()
59 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in TransferLiteralToInfeed()
2057 return ShapeUtil::ForEachSubshapeWithStatus( in CheckWhileBuffersShareAllocation()2086 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckConditionalBuffersShareAllocation()2097 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in CheckConditionalBuffersShareAllocation()
2304 TF_RETURN_IF_ERROR(ShapeUtil::ForEachSubshapeWithStatus( in HandleWhile()
523 static Status ForEachSubshapeWithStatus(const Shape& shape,
918 /* static */ Status ShapeUtil::ForEachSubshapeWithStatus( in ForEachSubshapeWithStatus() function in xla::ShapeUtil