Home
last modified time | relevance | path

Searched refs:ForEachMutableSubshapeWithStatus (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dutils.cc174 return ShapeUtil::ForEachMutableSubshapeWithStatus( in DetermineArgumentLayoutsFromCompileOptions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.h488 static Status ForEachMutableSubshapeWithStatus(
H A Dshape_util.cc1130 /* static */ Status ShapeUtil::ForEachMutableSubshapeWithStatus( in ForEachMutableSubshapeWithStatus() function in xla::ShapeUtil
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dxrt_state_ops.h579 Status status = xla::ShapeUtil::ForEachMutableSubshapeWithStatus( in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment.cc1997 TF_RETURN_IF_ERROR(ShapeUtil::ForEachMutableSubshapeWithStatus( in AssignLayouts()