Home
last modified time | relevance | path

Searched defs:xla_shapes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Doutfeed_ops.cc65 std::vector<xla::Shape> xla_shapes; in Compile() local
Dhost_compute_ops.cc51 std::vector<xla::Shape>* xla_shapes, in MakeXlaShapes()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dinfeed_ops.cc305 std::vector<xla::Shape> xla_shapes; in PrelinearizeTupleOp() local
478 std::vector<xla::Shape> xla_shapes; in TpuInfeedEnqueueTupleOp() local
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h276 std::vector<xla::Shape> xla_shapes; in XRTAllocateFromTensorOp() local