Searched refs:shape_ptr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 96 tensorflow::string ShapeString(const void* shape_ptr, xla::int32 shape_length) { in ShapeString() argument
98 xla::llvm_ir::DecodeSelfDescribingShapeConstant(shape_ptr, shape_length); in ShapeString()
134 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() argument
139 << ShapeString(shape_ptr, shape_length) << " on stream executor " in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
145 xla::llvm_ir::DecodeSelfDescribingShapeConstant(shape_ptr, shape_length); in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
153 const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() argument
158 << ShapeString(shape_ptr, shape_length) << " on stream executor " in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
170 << ShapeString(shape_ptr, shape_length); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
177 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() argument
182 << ShapeString(shape_ptr, shape_length) << " on stream executor " in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
[all …]
|
| D | cpu_runtime.h | 124 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length);
130 const void* shape_ptr, xla::int32 shape_length);
143 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length);
|
| D | ir_emitter.cc | 397 llvm::Value * shape_ptr, in EmitXfeedTransfer()
456 shape_ptr, b_.getInt32(shape_length)}); in EmitXfeedTransfer()
469 acquired_pointer, shape_ptr, b_.getInt32(shape_length)}); in EmitXfeedTransfer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_util.h | 149 StatusOr<Shape> DecodeSelfDescribingShapeConstant(const void* shape_ptr,
|
| D | llvm_util.cc | 248 StatusOr<Shape> DecodeSelfDescribingShapeConstant(const void* shape_ptr, in DecodeSelfDescribingShapeConstant() argument
251 TF_RET_CHECK(shape_proto.ParseFromArray(shape_ptr, size_bytes)); in DecodeSelfDescribingShapeConstant()
|