Lines Matching refs:xla
16 #include "tensorflow/compiler/xla/service/cpu/cpu_runtime.h"
20 #include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
25 namespace xla {
65 } // namespace xla
69 tensorflow::string ShapeString(const void* shape_ptr, xla::int32 shape_length) {
70 xla::StatusOr<xla::Shape> shape =
71 xla::llvm_ir::DecodeSelfDescribingShapeConstant(shape_ptr, shape_length);
73 return xla::ShapeUtil::HumanStringWithLayout(shape.ValueOrDie());
80 void* __xla_cpu_runtime_AcquireInfeedBufferForDequeue(xla::int32 buffer_length,
82 xla::int32 shape_length) {
87 xla::cpu::runtime::XfeedManager* xfeed = xla::cpu::runtime::GetXfeedManager();
89 xla::cpu::runtime::XfeedBuffer* buffer =
92 << "XLA program infeed request buffer size " << buffer_length
100 xla::int32 buffer_length, void* buffer_ptr, const void* shape_ptr,
101 xla::int32 shape_length) {
106 xla::cpu::runtime::XfeedManager* xfeed = xla::cpu::runtime::GetXfeedManager();
107 xla::StatusOr<xla::Shape> shape =
108 xla::llvm_ir::DecodeSelfDescribingShapeConstant(shape_ptr, shape_length);
114 xla::int32 buffer_length, const void* shape_ptr, xla::int32 shape_length) {
119 xla::cpu::runtime::XfeedManager* xfeed = xla::cpu::runtime::GetXfeedManager();
121 xla::cpu::runtime::XfeedBuffer* buffer =
124 << "XLA program outfeed request buffer size " << buffer_length
132 xla::int32 buffer_length, void* buffer_ptr, const void* shape_ptr,
133 xla::int32 shape_length) {
138 xla::cpu::runtime::XfeedManager* xfeed = xla::cpu::runtime::GetXfeedManager();
139 xla::StatusOr<xla::Shape> shape =
140 xla::llvm_ir::DecodeSelfDescribingShapeConstant(shape_ptr, shape_length);
