Home
last modified time | relevance | path

Searched refs:shape_length (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.h129 const void* shape, xla::int32 shape_length);
145 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length);
151 const void* shape_ptr, xla::int32 shape_length);
164 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length);
177 const void* shape_ptr, xla::int32 shape_length, xla::int32 num_buffers,
Dcpu_runtime.cc202 tensorflow::string ShapeString(const void* shape_ptr, xla::int32 shape_length) { in ShapeString() argument
204 DecodeSelfDescribingShapeConstant(shape_ptr, shape_length); in ShapeString()
234 const void* shape, xla::int32 shape_length) { in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() argument
239 << ShapeString(shape, shape_length) << " on stream executor " in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
251 << ShapeString(shape, shape_length); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
258 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() argument
263 << ShapeString(shape_ptr, shape_length) << " on stream executor " in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
269 DecodeSelfDescribingShapeConstant(shape_ptr, shape_length); in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
277 const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() argument
282 << ShapeString(shape_ptr, shape_length) << " on stream executor " in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
[all …]
Dir_emitter.cc459 int32 shape_length; in EmitXfeedTransfer() local
462 llvm_ir::EncodeSelfDescribingShapeConstant(shape, &shape_length, &b_)); in EmitXfeedTransfer()
478 shape_ptr, b_.getInt32(shape_length)}, in EmitXfeedTransfer()
509 acquired_pointer, shape_ptr, b_.getInt32(shape_length)}, in EmitXfeedTransfer()
1170 int32 shape_length; in HandleAllReduceMultipleReplica() local
1173 crs->shape(), &shape_length, &b_)); in HandleAllReduceMultipleReplica()
1192 /*shape_length=*/b_.getInt32(shape_length), in HandleAllReduceMultipleReplica()