Searched defs:shape_ptr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 189 const void* shape_ptr, xla::int32 size_bytes) { in DecodeSelfDescribingShapeConstant()
202 tensorflow::string ShapeString(const void* shape_ptr, xla::int32 shape_length) { in ShapeString()
258 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
277 const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
301 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
663 const void* shape_ptr, xla::int32 shape_length, xla::int32 num_buffers, in __xla_cpu_runtime_AllReduce()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.cc | 156 const int64* const shape_ptr = shape_.data(); in IndicesValidMatrix32BitFastPath() local
227 const int64* const shape_ptr = shape_.data(); in IndicesValidHelper() local
|