Home
last modified time | relevance | path

Searched defs:output_i (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions_test.cc62 int output_i = i - r; in TEST() local
114 int output_i = i - r; in TEST() local
162 int output_i = i - r + (patch_rows - 1) / 2; in TEST() local
213 int output_i = i - r + (patch_rows - 1) / 2; in TEST() local
267 int output_i = i - r; in TEST() local
323 int output_i = i - r; in TEST() local
377 int output_i = i - r; in test_batched_strided_spatial_convolution_backward_input_valid() local
454 int output_i = i - r; in test_batched_strided_spatial_convolution_backward_input_valid_row_major() local
529 int output_i = i - r; in TEST() local
585 int output_i = i - r; in TEST() local
[all …]
H A Deigen_backward_cuboid_convolutions_test.cc73 int output_i = i - p; in TEST() local
136 int output_i = i - p; in TEST() local
205 int output_i = i - p + forward_pad_z; in TEST() local
276 int output_i = i - p + forward_pad_z; in TEST() local
342 int output_i = i - p; in TEST() local
409 int output_i = i - p; in TEST() local
478 int output_i = i - p; in TEST() local
544 int output_i = i - p; in TEST() local
610 int output_i = i - p; in TEST() local
677 int output_i = i - p; in TEST() local
[all …]
H A Dsparse_fill_empty_rows_op_gpu.cu.cc137 Tindex output_i = i + num_new_rows_before[row]; in ScatterInputElementsKernel() local
158 Tindex output_i = input_i + (row == 0 ? 0 : num_new_rows_through[row - 1]); in ScatterNewElementsKernel() local
493 Tindex output_i = reverse_index_map[input_i]; in GatherOriginalGradValuesKernel() local
H A Dsparse_fill_empty_rows_op.cc169 const Tindex output_i = ((row == 0) ? 0 : csr_offset[row - 1]) + offset; in operator ()() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/
H A Ddepthwise_depthfirst.hpp370 unsigned int output_i, unsigned int output_j, in compute_tile_padded()
419 const unsigned int output_i, unsigned int output_j, unsigned int n_tile_cols, in compute_row_padded_tile_row()
495 unsigned int output_i, const unsigned int output_j, in compute_tiles_unpadded()
H A Ddepthfirst_driver.hpp96 const unsigned int output_i, unsigned int output_j, unsigned int n_tile_cols, in compute_row_padded_tile_row()
H A Ddepthwise_depthfirst_generic.hpp251 unsigned int output_i, unsigned int output_j, in compute_tile_padded()
H A Ddepthwise_depthfirst_multiplier.hpp528 unsigned int output_i, unsigned int output_j, in compute_tile_padded()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/
H A Dpooling_depthfirst.hpp130 unsigned int output_i, unsigned int output_j, in compute_tile_padded()
192 const unsigned int output_i, unsigned int output_j, unsigned int n_tile_cols, in compute_row_padded_tile_row()
H A Dpooling_depthfirst_generic.hpp145 unsigned int output_i, unsigned int output_j, in compute_tile_padded()
207 const unsigned int output_i, unsigned int output_j, unsigned int n_tile_cols, in compute_row_padded_tile_row()
H A Ddepthfirst_driver.hpp87 const unsigned int output_i, unsigned int output_j, unsigned int n_tile_cols, in compute_row_padded_tile_row()
H A Dpooling_depthfirst_cache_oblivious.hpp172 unsigned int output_i, output_j; in execute() member
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dfixup_onnx_controlflow.cpp611 Value* output_i = node->output(i); in ONNXMergeIfBlockOutputShapes() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/
H A DVulkanBackend.cpp446 const size_t output_i, in maybe_resize_output()
/aosp_15_r20/external/mesa3d/src/imgui/
H A Dimgui_draw.cpp1815 for (int output_i = 0; output_i < atlas->Fonts.Size && src_tmp.DstIndex == -1; output_i++) in ImFontAtlasBuildWithStbTruetype() local