Home
last modified time | relevance | path

Searched defs:input_ptrs (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddynamic_stitch_op_gpu.cu.cc35 GpuDeviceArrayStruct<const T*> input_ptrs, in DynamicStitchKernel()
55 const GpuDeviceArrayStruct<const T*>& input_ptrs, in DynamicStitchGPUImpl()
Dconcat_lib_gpu_impl.cu.cc40 const T** input_ptrs = GetGpuDeviceArrayOnDevice(&input_ptr_data); in concat_fixed_kernel() local
65 const T** input_ptrs = GetGpuDeviceArrayOnDevice(&input_ptr_data); in concat_variable_kernel() local
139 const GpuDeviceArrayStruct<const T*>& input_ptrs, in ConcatGPUImpl()
Dconcat_lib_gpu.cc41 GpuDeviceArrayOnHost<const T*> input_ptrs(c, inputs_flat.size()); in ConcatGPUCall() local
/external/ComputeLibrary/src/core/NEON/kernels/
DNENonLinearFilterKernel.cpp523 std::array<const unsigned char *, mask_h> input_ptrs{ {} }; in min_filter_box() local
562 std::array<const unsigned char *, mask_h> input_ptrs{ {} }; in max_filter_box() local
670 std::array<const unsigned char *, mask_h> input_ptrs{ {} }; in min_filter_cross() local
716 std::array<unsigned char *, mask_h> input_ptrs{ {} }; in max_filter_cross() local
886 std::array<unsigned char *, mask_h> input_ptrs{ {} }; in non_linear_filter_generic() local
DNEConvolutionKernel.cpp1204 std::array<unsigned char *, matrix_size> input_ptrs{ {} }; in convolution_u16() local
1270 std::array<unsigned char *, matrix_size> input_ptrs{ {} }; in convolution_s16() local
1336 std::array<unsigned char *, matrix_size> input_ptrs{ {} }; in convolution_s32() local
1569 std::array<unsigned char *, rows> input_ptrs{ {} }; in convolution() local
DNEPoolingLayerKernel.cpp1837 std::array<const uint8_t *, pool_size> input_ptrs{ {} }; in pooling7_f32_nchw() local
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dinference_profiler_stage.cc140 std::vector<void*> input_ptrs; in Run() local