Home
last modified time | relevance | path

Searched defs:GetTPUInfeedLayout (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/
H A Dinfeed_op.cc34 xla::Shape GetTPUInfeedLayout(const xla::Shape& shape) { in GetTPUInfeedLayout() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dadjust_layout.cc65 FailureOr<Attribute> GetTPUInfeedLayout(const ArrayRef<Type> types, in GetTPUInfeedLayout() function
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dinfeed_ops.cc57 xla::Shape GetTPUInfeedLayout(const xla::Shape& shape) { in GetTPUInfeedLayout() function