Home
last modified time | relevance | path

Searched defs:input_i (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/
H A Ddepthwise_depthfirst.hpp386 const auto input_i = static_cast<unsigned int>(ii < 0 ? 0 : ii); in compute_tile_padded() local
437 const auto input_i = static_cast<unsigned int>(ii < 0 ? 0 : ii); in compute_row_padded_tile_row() local
542 …const int input_i = static_cast<int>(output_i * this->m_args.stride_rows) - this->m_args.padding.t… in compute_tiles_unpadded() local
H A Ddepthwise_depthfirst_generic.hpp264 const auto input_i = static_cast<unsigned int>(ii < 0 ? 0 : ii); in compute_tile_padded() local
H A Ddepthwise_planar.hpp385 const unsigned int input_i = start_input_i < 0 ? 0 : start_input_i; in execute_internal() local
H A Ddepthwise_depthfirst_multiplier.hpp541 const auto input_i = static_cast<unsigned int>(ii < 0 ? 0 : ii); in compute_tile_padded() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/
H A Dpooling_depthfirst.hpp150 const auto input_i = static_cast<unsigned int>(ii < 0 ? 0 : ii); in compute_tile_padded() local
212 const auto input_i = static_cast<unsigned int>(ii < 0 ? 0 : ii); in compute_row_padded_tile_row() local
H A Dpooling_depthfirst_generic.hpp154 const auto input_i = static_cast<unsigned int>(start_i < 0 ? 0 : start_i); in compute_tile_padded() local
217 const auto input_i = static_cast<unsigned int>(start_i < 0 ? 0 : start_i); in compute_row_padded_tile_row() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_fill_empty_rows_op_gpu.cu.cc135 Tindex input_i = input_index_map ? input_index_map[i] : i; in ScatterInputElementsKernel() local
157 Tindex input_i = (row == 0 ? 0 : input_row_ends[row - 1]); in ScatterNewElementsKernel() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dfixup_onnx_controlflow.cpp252 Value* input_i = block->inputs().at(i); in FixupONNXLoopBlockInputs() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/
H A DVulkanBackend.cpp391 const size_t input_i, in maybe_resize_input()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc455 const string& input_i = root_node->input(i); in CreateOptimizedNodesGroup() local