Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.h72 Status GetShapeWithLayout(
Dshape_util.cc173 Status GetShapeWithLayout( in GetShapeWithLayout() function
/external/tensorflow/tensorflow/core/tpu/kernels/
Dinfeed_ops.cc150 return GetShapeWithLayout(input_shape, minor_to_major, layout_func, in GetInfeedShapeWithLayout()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h290 OP_REQUIRES_OK(ctx, GetShapeWithLayout(shape_, minor_to_major, in XRTAllocateFromTensorOp()