Home
last modified time | relevance | path

Searched refs:GetShapeWithLayout (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dshape_util.h75 Status GetShapeWithLayout(
H A Dshape_util.cc172 Status GetShapeWithLayout( in GetShapeWithLayout() function
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dinfeed_ops.cc150 return GetShapeWithLayout(input_shape, minor_to_major, layout_func, in GetInfeedShapeWithLayout()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dxrt_state_ops.h291 OP_REQUIRES_OK(ctx, GetShapeWithLayout(shape_, minor_to_major, in XRTAllocateFromTensorOp()