Home
last modified time | relevance | path

Searched refs:output_it (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dbroadcast.cc48 for (auto output_it = output_dims.rbegin(); output_it != output_dims.rend(); in BroadcastTo() local
49 ++output_it) { in BroadcastTo()
51 if (!(*output_it == 0 && *input_it == 0) && in BroadcastTo()
52 !(*input_it != 0 && *output_it % *input_it == 0)) { in BroadcastTo()
60 if (*output_it == *input_it || *input_it == 1) { in BroadcastTo()
61 broadcast_shape.push_back(*output_it); in BroadcastTo()
62 } else if (*output_it != *input_it) { in BroadcastTo()
67 broadcast_shape.push_back(*output_it / *input_it); in BroadcastTo()
71 broadcast_shape.push_back(*output_it); in BroadcastTo()
/external/ComputeLibrary/src/runtime/CPP/functions/
DCPPBoxWithNonMaximaSuppressionLimit.cpp43 Iterator output_it(output, window); in dequantize_tensor() local
50 …*reinterpret_cast<float *>(output_it.ptr()) = dequantize(*reinterpret_cast<const uint8_t *>(input_… in dequantize_tensor()
52 input_it, output_it); in dequantize_tensor()
57 …*reinterpret_cast<float *>(output_it.ptr()) = dequantize_qasymm8_signed(*reinterpret_cast<const in… in dequantize_tensor()
59 input_it, output_it); in dequantize_tensor()
64 …*reinterpret_cast<float *>(output_it.ptr()) = dequantize(*reinterpret_cast<const uint16_t *>(input… in dequantize_tensor()
66 input_it, output_it); in dequantize_tensor()
81 Iterator output_it(output, window); in quantize_tensor() local
88 …*reinterpret_cast<uint8_t *>(output_it.ptr()) = quantize_qasymm8(*reinterpret_cast<const float *>(… in quantize_tensor()
90 input_it, output_it); in quantize_tensor()
[all …]
/external/libtextclassifier/abseil-cpp/absl/strings/
Dascii.cc180 auto output_it = &(*str)[0]; in RemoveExtraAsciiWhitespace() local
187 if (is_ws) --output_it; in RemoveExtraAsciiWhitespace()
192 *output_it = *input_it; in RemoveExtraAsciiWhitespace()
193 ++output_it; in RemoveExtraAsciiWhitespace()
196 str->erase(output_it - &(*str)[0]); in RemoveExtraAsciiWhitespace()
/external/abseil-cpp/absl/strings/
Dascii.cc180 auto output_it = &(*str)[0]; in RemoveExtraAsciiWhitespace() local
187 if (is_ws) --output_it; in RemoveExtraAsciiWhitespace()
192 *output_it = *input_it; in RemoveExtraAsciiWhitespace()
193 ++output_it; in RemoveExtraAsciiWhitespace()
196 str->erase(output_it - &(*str)[0]); in RemoveExtraAsciiWhitespace()
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/strings/
Dascii.cc180 auto output_it = &(*str)[0]; in RemoveExtraAsciiWhitespace() local
187 if (is_ws) --output_it; in RemoveExtraAsciiWhitespace()
192 *output_it = *input_it; in RemoveExtraAsciiWhitespace()
193 ++output_it; in RemoveExtraAsciiWhitespace()
196 str->erase(output_it - &(*str)[0]); in RemoveExtraAsciiWhitespace()
/external/openscreen/third_party/abseil/src/absl/strings/
Dascii.cc180 auto output_it = &(*str)[0]; in RemoveExtraAsciiWhitespace() local
187 if (is_ws) --output_it; in RemoveExtraAsciiWhitespace()
192 *output_it = *input_it; in RemoveExtraAsciiWhitespace()
193 ++output_it; in RemoveExtraAsciiWhitespace()
196 str->erase(output_it - &(*str)[0]); in RemoveExtraAsciiWhitespace()
/external/tensorflow/third_party/absl/abseil-cpp/absl/strings/
Dascii.cc180 auto output_it = &(*str)[0]; in RemoveExtraAsciiWhitespace() local
187 if (is_ws) --output_it; in RemoveExtraAsciiWhitespace()
192 *output_it = *input_it; in RemoveExtraAsciiWhitespace()
193 ++output_it; in RemoveExtraAsciiWhitespace()
196 str->erase(static_cast<size_t>(output_it - &(*str)[0])); in RemoveExtraAsciiWhitespace()
/external/webrtc/third_party/abseil-cpp/absl/strings/
Dascii.cc180 auto output_it = &(*str)[0]; in RemoveExtraAsciiWhitespace() local
187 if (is_ws) --output_it; in RemoveExtraAsciiWhitespace()
192 *output_it = *input_it; in RemoveExtraAsciiWhitespace()
193 ++output_it; in RemoveExtraAsciiWhitespace()
196 str->erase(static_cast<size_t>(output_it - &(*str)[0])); in RemoveExtraAsciiWhitespace()
/external/angle/third_party/abseil-cpp/absl/strings/
Dascii.cc228 auto output_it = &(*str)[0]; in RemoveExtraAsciiWhitespace() local
235 if (is_ws) --output_it; in RemoveExtraAsciiWhitespace()
240 *output_it = *input_it; in RemoveExtraAsciiWhitespace()
241 ++output_it; in RemoveExtraAsciiWhitespace()
244 str->erase(static_cast<size_t>(output_it - &(*str)[0])); in RemoveExtraAsciiWhitespace()
/external/cronet/third_party/abseil-cpp/absl/strings/
Dascii.cc228 auto output_it = &(*str)[0]; in RemoveExtraAsciiWhitespace() local
235 if (is_ws) --output_it; in RemoveExtraAsciiWhitespace()
240 *output_it = *input_it; in RemoveExtraAsciiWhitespace()
241 ++output_it; in RemoveExtraAsciiWhitespace()
244 str->erase(static_cast<size_t>(output_it - &(*str)[0])); in RemoveExtraAsciiWhitespace()
/external/ComputeLibrary/src/cpu/kernels/l2normlayer/generic/neon/
Dimpl.cpp51 Iterator output_it(out, win_collapsed); in l2_normalize_x() local
56 const auto out_ptr = reinterpret_cast<T *>(output_it.ptr()); in l2_normalize_x()
75 input_it, sum_it, output_it); in l2_normalize_x()
95 Iterator output_it(out, win); in l2_normalize_yz() local
103 const auto out_ptr = reinterpret_cast<T *>(output_it.ptr()); in l2_normalize_yz()
120 input_it, sum_it, output_it); in l2_normalize_yz()
/external/ComputeLibrary/src/core/NEON/kernels/
DNEGatherKernel.cpp133 Iterator output_it(_output, window); in gather_0_axis() local
141 … std::copy_n(_input->ptr_to_element(gather_id), _output->info()->element_size(), output_it.ptr()); in gather_0_axis()
143 output_it); in gather_0_axis()
157 Iterator output_it(_output, output_window); in gather_n_axis() local
165 …ement(gather_id), _input->info()->dimension(0) * _output->info()->element_size(), output_it.ptr()); in gather_n_axis()
167 output_it); in gather_n_axis()
DNETileKernel.cpp105 Iterator output_it(_output, out_slice); in run() local
114 …memcpy(output_it.ptr(), _input->ptr_to_element(input_coords), _input->info()->dimension(0) * _inpu… in run()
116 output_it); in run()
DNEPadLayerKernel.cpp65 Iterator output_it(_output, output_window); in run_pad_constant() local
74 …std::fill_n(reinterpret_cast<T *>(output_it.ptr()), _output->info()->dimension(0), _constant_value… in run_pad_constant()
79 T *output_it_ptr = reinterpret_cast<T *>(output_it.ptr()); in run_pad_constant()
84 output_it); in run_pad_constant()
DNEStridedSliceKernel.cpp165 Iterator output_it(output, win); in run_op() local
194 std::copy_n(cur_ptr, width_size, output_it.ptr()); in run_op()
196 output_it); in run_op()
/external/ComputeLibrary/src/cpu/kernels/depthwiseconv2d/generic/neon/
Dimpl.cpp160 Iterator output_it(dst, win_output); in depthwise_loop_multiplier1_fp() local
208 wrapper::vstore(reinterpret_cast<T *>(output_it.ptr()) + x, acc); in depthwise_loop_multiplier1_fp()
240 *(reinterpret_cast<T *>(output_it.ptr()) + x) = acc_scalar; in depthwise_loop_multiplier1_fp()
243 input_it, weights_it, biases_it, output_it); in depthwise_loop_multiplier1_fp()
271 Iterator output_it(dst, win_output); in depthwise_loop_generic_fp() local
314 *(reinterpret_cast<T *>(output_it.ptr() + m * sizeof(T))) = acc.at(m) + biases_val; in depthwise_loop_generic_fp()
321 *(reinterpret_cast<T *>(output_it.ptr() + m * sizeof(T))) = acc.at(m); in depthwise_loop_generic_fp()
325 input_it, weights_it, biases_it, output_it); in depthwise_loop_generic_fp()
365 Iterator output_it(dst, win_output); in depthwise_loop_multiplier1_quantized() local
440 wrapper::vstore(reinterpret_cast<T *>(output_it.ptr()) + x, out_vals); in depthwise_loop_multiplier1_quantized()
[all …]
/external/ComputeLibrary/src/cpu/kernels/range/generic/neon/
Dimpl.cpp51 Iterator output_it(output, win); in neon_range_function() local
56 const auto out_ptr = reinterpret_cast<T *>(output_it.ptr()); in neon_range_function()
77 output_it); in neon_range_function()
/external/ComputeLibrary/examples/
Dneon_copy_objects.cpp117 Iterator output_it(&output, output_window); in do_setup() local
133 …memcpy(dst_data + id.z() * (width * height) + id.y() * width, output_it.ptr(), width * sizeof(floa… in do_setup()
135 output_it); in do_setup()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h712 auto output_it = src_output_idx_.find(inp.second); in input_shapes() local
713 if (output_it == src_output_idx_.end()) continue; in input_shapes()
718 const auto& output_shape = output_shapes.find(output_it->second); in input_shapes()
/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc3573 auto output_it = outputs_in_alternate_memory_map.find(instruction); in ComputeEstimatedElapsedTime() local
3574 if (output_it != outputs_in_alternate_memory_map.end()) { in ComputeEstimatedElapsedTime()
3575 outputs_in_alternate_memory = output_it->second; in ComputeEstimatedElapsedTime()