Home
last modified time | relevance | path

Searched refs:data_index (Results 1 – 25 of 27) sorted by relevance

12

/external/pytorch/aten/src/ATen/native/
DSegmentReduce.cpp90 int64_t data_index = outer_idx * data_stride_axis * data_size_axis in _segment_reduce_lengths_cpu_kernel1() local
92 const auto val = values_data[data_index]; in _segment_reduce_lengths_cpu_kernel1()
255 int64_t data_index = outer_idx * data_stride_axis * data_size_axis in _segment_reduce_cpu_lengths_backward_kernel1() local
257 if (at::_isnan(values_data[data_index]) || in _segment_reduce_cpu_lengths_backward_kernel1()
258 values_data[data_index] == output_data[output_index]) { in _segment_reduce_cpu_lengths_backward_kernel1()
259 grad_input_data[data_index] = grad_data[output_index]; in _segment_reduce_cpu_lengths_backward_kernel1()
269 int64_t data_index = outer_idx * data_stride_axis * data_size_axis in _segment_reduce_cpu_lengths_backward_kernel1() local
271 if (grad_input_data[data_index] > 0) { in _segment_reduce_cpu_lengths_backward_kernel1()
272 grad_input_data[data_index] = in _segment_reduce_cpu_lengths_backward_kernel1()
273 grad_input_data[data_index] / counter; in _segment_reduce_cpu_lengths_backward_kernel1()
[all …]
/external/pytorch/aten/src/ATen/native/cuda/
DSegmentReduce.cu119 int64_t data_index = outer_idx * data_stride_axis * data_size_axis in segment_reduce_forward_kernel() local
121 const auto data = values_data[data_index]; in segment_reduce_forward_kernel()
201 int64_t data_index = outer_idx * data_stride_axis * data_size_axis in segment_reduce_backward_kernel() local
203 if (at::_isnan(values_data[data_index]) || in segment_reduce_backward_kernel()
204 values_data[data_index] == output_data[output_index]) { in segment_reduce_backward_kernel()
205 grad_input_data[data_index] = grad_data[output_index]; in segment_reduce_backward_kernel()
215 int64_t data_index = outer_idx * data_stride_axis * data_size_axis in segment_reduce_backward_kernel() local
217 if (grad_input_data[data_index] > 0) { in segment_reduce_backward_kernel()
218 grad_input_data[data_index] = in segment_reduce_backward_kernel()
219 grad_input_data[data_index] / counter; in segment_reduce_backward_kernel()
[all …]
/external/angle/third_party/abseil-cpp/absl/crc/internal/
Dcrc_memcpy_x86_arm_combined.cc130 size_t data_index = i * kIntLoadsPerVec + j; in LargeTailCopy() local
132 int_data[data_index] = *(usrc + j); in LargeTailCopy()
134 int_data[data_index])}; in LargeTailCopy()
136 *(udst + j) = int_data[data_index]; in LargeTailCopy()
298 size_t data_index = j * kIntLoadsPerVec + k; in Compute() local
301 int_data[data_index] = *(usrc + i * kIntLoadsPerVec + k); in Compute()
303 int_data[data_index])}; in Compute()
306 *(udst + i * kIntLoadsPerVec + k) = int_data[data_index]; in Compute()
/external/cronet/tot/third_party/abseil-cpp/absl/crc/internal/
Dcrc_memcpy_x86_arm_combined.cc130 size_t data_index = i * kIntLoadsPerVec + j; in LargeTailCopy() local
132 int_data[data_index] = *(usrc + j); in LargeTailCopy()
134 int_data[data_index])}; in LargeTailCopy()
136 *(udst + j) = int_data[data_index]; in LargeTailCopy()
298 size_t data_index = j * kIntLoadsPerVec + k; in Compute() local
301 int_data[data_index] = *(usrc + i * kIntLoadsPerVec + k); in Compute()
303 int_data[data_index])}; in Compute()
306 *(udst + i * kIntLoadsPerVec + k) = int_data[data_index]; in Compute()
/external/abseil-cpp/absl/crc/internal/
Dcrc_memcpy_x86_arm_combined.cc130 size_t data_index = i * kIntLoadsPerVec + j; in LargeTailCopy() local
132 int_data[data_index] = *(usrc + j); in LargeTailCopy()
134 int_data[data_index])}; in LargeTailCopy()
136 *(udst + j) = int_data[data_index]; in LargeTailCopy()
298 size_t data_index = j * kIntLoadsPerVec + k; in Compute() local
301 int_data[data_index] = *(usrc + i * kIntLoadsPerVec + k); in Compute()
303 int_data[data_index])}; in Compute()
306 *(udst + i * kIntLoadsPerVec + k) = int_data[data_index]; in Compute()
/external/cronet/stable/third_party/abseil-cpp/absl/crc/internal/
Dcrc_memcpy_x86_arm_combined.cc130 size_t data_index = i * kIntLoadsPerVec + j; in LargeTailCopy() local
132 int_data[data_index] = *(usrc + j); in LargeTailCopy()
134 int_data[data_index])}; in LargeTailCopy()
136 *(udst + j) = int_data[data_index]; in LargeTailCopy()
298 size_t data_index = j * kIntLoadsPerVec + k; in Compute() local
301 int_data[data_index] = *(usrc + i * kIntLoadsPerVec + k); in Compute()
303 int_data[data_index])}; in Compute()
306 *(udst + i * kIntLoadsPerVec + k) = int_data[data_index]; in Compute()
/external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/crc/internal/
Dcrc_memcpy_x86_64.cc125 size_t data_index = i * kIntLoadsPerVec + j; in LargeTailCopy() local
127 int_data[data_index] = *(usrc + j); in LargeTailCopy()
129 static_cast<uint32_t>(crcs[region]), int_data[data_index]))}; in LargeTailCopy()
131 *(udst + j) = int_data[data_index]; in LargeTailCopy()
291 size_t data_index = j * kIntLoadsPerVec + k; in Compute() local
294 int_data[data_index] = *(usrc + i * kIntLoadsPerVec + k); in Compute()
296 static_cast<uint32_t>(crcs[region]), int_data[data_index]))}; in Compute()
299 *(udst + i * kIntLoadsPerVec + k) = int_data[data_index]; in Compute()
/external/cronet/tot/testing/perf/confidence/
Dcompare_blink_perf.cc145 size_t data_index; member
231 data[label.data_index]) { in main()
235 double mean_before = sum_before / data[label.data_index].size(); in main()
236 double mean_after = sum_after / data[label.data_index].size(); in main()
239 estimates[label.data_index]; in main()
Dpinpoint_ci.cc158 unsigned data_index = 0; in main() local
162 to_print.emplace_back(key, std::move(estimates[data_index])); in main()
163 ++data_index; in main()
/external/cronet/stable/testing/perf/confidence/
Dcompare_blink_perf.cc145 size_t data_index; member
231 data[label.data_index]) { in main()
235 double mean_before = sum_before / data[label.data_index].size(); in main()
236 double mean_after = sum_after / data[label.data_index].size(); in main()
239 estimates[label.data_index]; in main()
Dpinpoint_ci.cc158 unsigned data_index = 0; in main() local
162 to_print.emplace_back(key, std::move(estimates[data_index])); in main()
163 ++data_index; in main()
/external/tensorflow/tensorflow/examples/speech_commands/
Dinput_data.py273 self.data_index = {'validation': [], 'testing': [], 'training': []}
290 self.data_index[set_index].append({'label': word, 'file': wav_path})
302 silence_wav_path = self.data_index['training'][0]['file']
304 set_size = len(self.data_index[set_index])
307 self.data_index[set_index].append({
314 self.data_index[set_index].extend(unknown_index[set_index][:unknown_size])
317 random.shuffle(self.data_index[set_index])
502 return len(self.data_index[mode])
532 candidates = self.data_index[mode]
648 candidates = self.data_index[mode]
Dinput_data_test.py104 self.assertIn("training", audio_processor.data_index)
105 self.assertIn("validation", audio_processor.data_index)
106 self.assertIn("testing", audio_processor.data_index)
/external/trusty/arm-trusted-firmware/drivers/st/crypto/
Dstm32_pka.c361 int data_index; in write_eo_data() local
368 data_index = (int)data_size - 1; in write_eo_data()
374 while ((i < sizeof(tmp)) && (data_index >= 0)) { in write_eo_data()
375 tmp |= (uint64_t)(data[data_index]) << (UINT8_LEN * i); in write_eo_data()
377 data_index--; /* Move to just next most significat byte */ in write_eo_data()
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc1091 int64_t* data_index, string* result) { in PrintOneDim() argument
1092 if (*data_index >= limit) return; in PrintOneDim()
1097 if (*data_index >= limit) { in PrintOneDim()
1105 strings::StrAppend(result, PrintOneElement(data[(*data_index)++], false)); in PrintOneDim()
1112 if (*data_index < limit) { in PrintOneDim()
1117 PrintOneDim(dim_index + 1, shape, limit, shape_size, data, data_index, in PrintOneDim()
1119 if (*data_index < limit || flag) { in PrintOneDim()
1144 int64_t data_index, string* result) { in PrintOneDimV2() argument
1148 strings::StrAppend(result, PrintOneElement(data[data_index], true)); in PrintOneDimV2()
1169 data_index + elements_per_iter * i, result); in PrintOneDimV2()
[all …]
/external/cronet/tot/third_party/protobuf/src/google/protobuf/compiler/cpp/
Denum.cc352 int data_index = 0; in GenerateMethods() local
354 format(" { {$classname$_names + $1$, $2$}, $3$ },\n", data_index, in GenerateMethods()
360 data_index += p.first.size(); in GenerateMethods()
/external/protobuf/src/google/protobuf/compiler/cpp/
Denum.cc352 int data_index = 0; in GenerateMethods() local
354 format(" { {$classname$_names + $1$, $2$}, $3$ },\n", data_index, in GenerateMethods()
360 data_index += p.first.size(); in GenerateMethods()
/external/cronet/stable/third_party/protobuf/src/google/protobuf/compiler/cpp/
Denum.cc352 int data_index = 0; in GenerateMethods() local
354 format(" { {$classname$_names + $1$, $2$}, $3$ },\n", data_index, in GenerateMethods()
360 data_index += p.first.size(); in GenerateMethods()
/external/mesa3d/src/imagination/vulkan/winsys/powervr/
Dpvr_drm_job_render.c497 .data_index = submit_info->rt_data_idx, in pvr_drm_winsys_render_submit()
510 .data_index = submit_info->rt_data_idx, in pvr_drm_winsys_render_submit()
578 .data_index = submit_info->rt_data_idx, in pvr_drm_winsys_render_submit()
/external/flashrom/
Draiden_debug_spi.c556 uint16_t data_index; member
1080 continue_packet.packet_v2.cmd_continue.data_index = in write_command_v2()
1139 response.packet_v2.rsp_continue.data_index) { in read_response_v2()
1141 response.packet_v2.rsp_continue.data_index, in read_response_v2()
/external/executorch/exir/_serialize/
D_program.py548 data_index: int = len(program.backend_delegate_data)
553 location=DataLocation.INLINE, index=data_index
/external/mesa3d/include/drm-uapi/
Dpvr_drm.h1223 __u32 data_index; member
/external/kernel-headers/original/uapi/drm/
Dpvr_drm.h1223 __u32 data_index; member
/external/executorch/exir/_serialize/test/
Dtest_program.py58 data_index: int = len(program.backend_delegate_data)
68 index=data_index,
/external/webrtc/pc/
Dsdp_offer_answer.cc3949 absl::optional<size_t> data_index; in GetOptionsForPlanBOffer() local
3958 &audio_index, &video_index, &data_index, session_options); in GetOptionsForPlanBOffer()
3995 if (!data_index && offer_new_data_description) { in GetOptionsForPlanBOffer()
4195 absl::optional<size_t> data_index; in GetOptionsForPlanBAnswer() local
4204 &video_index, &data_index, session_options); in GetOptionsForPlanBAnswer()
5020 absl::optional<size_t>* data_index, in GenerateMediaDescriptionOptions() argument
5068 if (*data_index) { in GenerateMediaDescriptionOptions()
5074 *data_index = session_options->media_description_options.size() - 1; in GenerateMediaDescriptionOptions()

12