Searched defs:shape_ptr (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_runtime.cc | 237 const void* shape_ptr, int32_t size_bytes) { in DecodeSelfDescribingShapeConstant()
250 std::string ShapeString(const void* shape_ptr, int32_t shape_length) { in ShapeString()
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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/ |
| H A D | sparse_tensor.cc | 156 const int64_t* const shape_ptr = shape_.data(); in IndicesValidMatrix32BitFastPath() local
227 const int64_t* const shape_ptr = shape_.data(); in IndicesValidHelper() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/backends/nnapi/ |
| H A D | nnapi_backend_lib.cpp | 118 auto shape_ptr = dict.at("shape_compute_module").toString(); in init() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_to_dense_op_gpu.cu.cc | 151 const Index* shape_ptr = shape.flat<Index>().data(); in operator ()() local
|