| /external/tensorflow/tensorflow/compiler/xla/ |
| D | array.h | 105 : sizes_(sizes.begin(), sizes.end()), values_(new T[num_elements()]) {
252 : sizes_(other.sizes_), values_(new T[num_elements()]) {
258 sizes_ = other.sizes_;
319 std::vector<int64> index(sizes_.size());
327 std::vector<int64> index(sizes_.size());
337 std::vector<int64> index(sizes_.size());
351 std::vector<int64> index(sizes_.size());
412 const int64 sizes_size = sizes_.size();
414 return sizes_[n];
418 const std::vector<int64>& dimensions() const { return sizes_; }
[all …]
|
| /external/pigweed/pw_allocator/public/pw_allocator/ |
| D | freelist.h | 97 : chunks_(chunks), sizes_(sizes) {} in FreeList()
100 Vector<size_t>& sizes_; variable
110 : FreeList(chunks_, sizes_), sizes_(sizes), chunks_(kNumBuckets + 1, 0) {} in FreeListBuffer()
112 : FreeList(chunks_, sizes_), in FreeListBuffer()
113 sizes_(sizes.begin(), sizes.end()), in FreeListBuffer()
117 Vector<size_t, kNumBuckets> sizes_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.h | 92 sizes_(sizes), in size_function_()
146 RematerializationSizes* sizes_; variable
|
| D | hlo_rematerialization.cc | 2097 if (sizes_ != nullptr) { in Run()
2098 sizes_->before_bytes = before_peak_memory; in Run()
2099 sizes_->after_bytes = current_peak_memory; in Run()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 254 mkldnn_dims_t sizes_; // Required by MKL for conversions member
270 for (size_t i = 0; i < sizeof(data_.sizes_) / sizeof(data_.sizes_[0]); in MklDnnShape()
272 data_.sizes_[i] = -1; in MklDnnShape()
371 return reinterpret_cast<const int*>(&data_.sizes_[0]); in GetSizes()
379 size_t dimensions = sizeof(data_.sizes_) / sizeof(data_.sizes_[0]); in GetSizesAsMklDnnDims()
381 if (data_.sizes_[i] != INVALID_DIM_SIZE) in GetSizesAsMklDnnDims()
382 retVal.push_back(data_.sizes_[i]); in GetSizesAsMklDnnDims()
391 CHECK_LT(index, sizeof(data_.sizes_) / sizeof(data_.sizes_[0])); in DimSize()
392 return data_.sizes_[index]; in DimSize()
405 shape[idx] = data_.sizes_[TfDimIdx(idx)]; in GetTfShape()
[all …]
|
| /external/pigweed/pw_allocator/ |
| D | freelist.cc | 117 for (chunk_ptr = 0; chunk_ptr < sizes_.size(); chunk_ptr++) { in FindChunkPtrForSize()
118 if (sizes_[chunk_ptr] >= size && in FindChunkPtrForSize()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | api.cc | 92 : sizes_(sizes), in InferenceContextWithBatchImpl()
107 for (const auto& s : sizes_) { in Execute()
131 for (const auto& s : sizes_) { in Execute()
159 const ObjectsSizes sizes_; member in tflite::gpu::gl::__anon1290dc7a0111::InferenceContextWithBatchImpl
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 420 : index_nodes_(num_devices), sizes_(types.size()) { in TensorDevicePlacer()
424 sizes_[i] = GetInferredShapeSize(shapes[i], types[i]); in TensorDevicePlacer()
425 if (sizes_[i] >= 0) { in TensorDevicePlacer()
426 total_size += sizes_[i]; in TensorDevicePlacer()
435 for (int64 i = 0; i < sizes_.size(); ++i) { in TensorDevicePlacer()
436 if (sizes_[i] < 0) { in TensorDevicePlacer()
437 sizes_[i] = undefined_shape_size; in TensorDevicePlacer()
450 node->size += sizes_.at(index); in ReportDeviceAssigned()
459 node->size += sizes_.at(index); in RetrieveAssignment()
487 std::vector<int64> sizes_; member in tensorflow::__anonf0ad67bd0111::TensorDevicePlacer
|
| /external/mesa3d/include/CL/ |
| D | cl2.hpp | 5691 size_type sizes_[3]; member in cl::NDRange
5699 sizes_[0] = 0; in NDRange()
5700 sizes_[1] = 0; in NDRange()
5701 sizes_[2] = 0; in NDRange()
5708 sizes_[0] = size0; in NDRange()
5709 sizes_[1] = 1; in NDRange()
5710 sizes_[2] = 1; in NDRange()
5717 sizes_[0] = size0; in NDRange()
5718 sizes_[1] = size1; in NDRange()
5719 sizes_[2] = 1; in NDRange()
[all …]
|
| D | cl.hpp | 4769 size_t<3> sizes_; member in cl::NDRange
4782 sizes_[0] = size0; in NDRange()
4789 sizes_[0] = size0; in NDRange()
4790 sizes_[1] = size1; in NDRange()
4797 sizes_[0] = size0; in NDRange()
4798 sizes_[1] = size1; in NDRange()
4799 sizes_[2] = size2; in NDRange()
4807 return (const ::size_t*) sizes_; in operator const::size_t*()
|