Searched defs:shape_length (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_runtime.cc | 250 std::string ShapeString(const void* shape_ptr, int32_t shape_length) { in ShapeString()
302 const void* shape, int32_t shape_length) { in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
325 void* buffer_ptr, const void* shape_ptr, int32_t shape_length) { in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
343 const void* shape_ptr, int32_t shape_length) { in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
366 void* buffer_ptr, const void* shape_ptr, int32_t shape_length) { in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
764 int32_t reduction_kind, const void* shape_ptr, int32_t shape_length, in __xla_cpu_runtime_AllReduce()
|
| H A D | ir_emitter.cc | 473 int32_t shape_length; in EmitXfeedTransfer() local
1244 int32_t shape_length; in HandleAllReduceMultipleReplica() local
|