| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | Conv_v8.cpp | 369 void* data_ptrs[3]; in run_conv_plan() local
372 data_ptrs[0] = const_cast<void*>(x.const_data_ptr()); in run_conv_plan()
373 data_ptrs[1] = y.data_ptr(); in run_conv_plan()
374 data_ptrs[2] = const_cast<void*>(w.const_data_ptr()); in run_conv_plan()
378 data_ptrs[0] = x.data_ptr(); in run_conv_plan()
379 data_ptrs[1] = const_cast<void*>(y.const_data_ptr()); in run_conv_plan()
380 data_ptrs[2] = const_cast<void*>(w.const_data_ptr()); in run_conv_plan()
384 data_ptrs[0] = const_cast<void*>(x.const_data_ptr()); in run_conv_plan()
385 data_ptrs[1] = const_cast<void*>(y.const_data_ptr()); in run_conv_plan()
386 data_ptrs[2] = w.data_ptr(); in run_conv_plan()
[all …]
|
| D | RNN.cpp | 894 std::vector<void*> data_ptrs; in get_parameters() local
896 data_ptrs.reserve(num_dir_layers * (2 * 2 + 1)); in get_parameters()
898 data_ptrs.reserve(num_dir_layers * 2 * 2); in get_parameters()
948 data_ptrs.push_back(matrix_pointer); in get_parameters()
983 data_ptrs.push_back(matrix_pointer); in get_parameters()
986 return data_ptrs; in get_parameters()
|
| /external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| D | BinaryOps.cpp | 140 std::vector<void *> data_ptrs; in add() local
142 data_ptrs.reserve(8); in add()
144 data_ptrs = {qb.data_ptr<int8_t>(), rhs_multiplier_tensor.data_ptr(), add_output.data_ptr(), in add()
149 data_ptrs.emplace_back(add_output.data_ptr()), in add()
155 .setDataPointers(static_cast<int64_t>(uids.size()), data_ptrs.data()) in add()
|
| D | Linear.cpp | 148 at::SmallVector<void *, 8> data_ptrs; in apply_impl_helper() local
150 data_ptrs = {input.data_ptr<int8_t>(), weight_transposed.data_ptr<int8_t>(), in apply_impl_helper()
154 …data_ptrs.insert(data_ptrs.end(), {broadcasted_bias.value().data_ptr(), bias_multiplier_tensor.val… in apply_impl_helper()
160 .setDataPointers(static_cast<int64_t>(uids.size()), data_ptrs.data()) in apply_impl_helper()
|
| D | Conv.cpp | 120 at::SmallVector<void *, 7> data_ptrs; in apply_impl_helper() local
122 data_ptrs = {input.data_ptr<int8_t>(), maybe_padded_weight_.data_ptr<int8_t>(), in apply_impl_helper()
126 …data_ptrs.insert(data_ptrs.end(), {broadcasted_bias.value().data_ptr(), bias_multiplier_tensor.val… in apply_impl_helper()
132 .setDataPointers(static_cast<int64_t>(uids.size()), data_ptrs.data()) in apply_impl_helper()
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
| D | object_detection_stage.cc | 107 std::vector<void*> data_ptrs = {}; in Run() local
108 data_ptrs.push_back(preprocessing_stage_->GetPreprocessedImageData()); in Run()
109 inference_stage_->SetInputs(data_ptrs); in Run()
|
| D | image_classification_stage.cc | 117 std::vector<void*> data_ptrs = {}; in Run() local
118 data_ptrs.push_back(preprocessing_stage_->GetPreprocessedImageData()); in Run()
119 inference_stage_->SetInputs(data_ptrs); in Run()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op_gpu.cu.cc | 38 const T** data_ptrs = GetGpuDeviceArrayOnDevice(&input_ptrs); in DynamicStitchKernel() local
44 output[output_index] = ldg(data_ptrs[input_index] + slice_offset); in DynamicStitchKernel()
|
| /external/libaom/test/ |
| D | noise_model_test.cc | 1269 const uint8_t *const data_ptrs[3] = { in TYPED_TEST_P() local
1279 EXPECT_EQ(0, aom_wiener_denoise_2d(data_ptrs, denoised_ptrs, this->kWidth, in TYPED_TEST_P()
1283 EXPECT_EQ(0, aom_wiener_denoise_2d(data_ptrs, denoised_ptrs, this->kWidth, in TYPED_TEST_P()
1287 EXPECT_EQ(0, aom_wiener_denoise_2d(data_ptrs, denoised_ptrs, this->kWidth, in TYPED_TEST_P()
1294 const uint8_t *const data_ptrs[3] = { in TYPED_TEST_P() local
1305 EXPECT_EQ(0, aom_wiener_denoise_2d(data_ptrs, denoised_ptrs, this->kWidth, in TYPED_TEST_P()
1312 EXPECT_EQ(0, aom_wiener_denoise_2d(data_ptrs, denoised_ptrs, this->kWidth, in TYPED_TEST_P()
1322 const uint8_t *const data_ptrs[3] = { in TYPED_TEST_P() local
1333 data_ptrs, denoised_ptrs, width, height, this->stride_, this->chroma_sub_, in TYPED_TEST_P()
|
| /external/pytorch/torch/onnx/_internal/ |
| D | onnxruntime.py | 368 data_ptrs = []
373 data_ptrs.append(tensor.data_ptr())
374 ortvalues.push_back_batch(tensors, data_ptrs, dtypes, shapes, devices)
|
| /external/pytorch/torch/csrc/cuda/ |
| D | Module.cpp | 1254 [](py::list& tensors, py::list& data_ptrs, py::list& indices) { in registerCudaPluggableAllocator() argument
1258 auto data_ptr = data_ptrs[index].cast<int64_t>(); in registerCudaPluggableAllocator()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| D | QuantizedOpKernels.cpp | 71 std::vector<void*> data_ptrs; in qcat_nhwc_kernel() local
102 data_ptrs.push_back(qx.data_ptr()); in qcat_nhwc_kernel()
138 reinterpret_cast<scalar_t::underlying*>(data_ptrs[tidx]) + in qcat_nhwc_kernel()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 4464 std::array<void*, sizeof...(Args)> data_ptrs = {inputs.opaque()...}; in operator ()() local
4469 data_ptrs.cbegin(), data_ptrs.cend()); in operator ()()
|