Searched defs:shape_length (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_runtime.cc | 215 tensorflow::string ShapeString(const void* shape_ptr, xla::int32 shape_length) { in ShapeString() 268 const void* shape, xla::int32 shape_length) { in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() 291 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() 309 const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() 332 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() 685 const void* shape_ptr, xla::int32 shape_length, xla::int32 num_buffers, in __xla_cpu_runtime_AllReduce()
|
D | ir_emitter.cc | 459 int32_t shape_length; in EmitXfeedTransfer() local 1172 int32_t shape_length; in HandleAllReduceMultipleReplica() local
|