Home
last modified time | relevance | path

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

12

/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/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst/
Dgeneric.cpp34 const float *const *const input_ptrs, in sve_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in sve_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_indirect_impl()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
Dbenchmark.h80 llvm::SmallVector<void*> input_ptrs; member
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst/
Dgeneric_indirect.cpp34 const __fp16 *const *const input_ptrs, in a64_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in a64_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const __fp16 *const *const input_ptrs, in a64_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_5x5_s1_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in a64_fp32_nhwc_5x5_s1_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in a64_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst/
Dgeneric_indirect.cpp34 const __fp16 *const *const input_ptrs, in sve_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in sme2_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in sve_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in sme2_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in a64_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const __fp16 *const *const input_ptrs, in sve_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in sme2_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in sme2_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in a64_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in sve_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const __fp16 *const *const input_ptrs, in sve_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst/
Dgeneric_indirect.cpp34 const float *const *const input_ptrs, in sve_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const __fp16 *const *const input_ptrs, in a64_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst_indirect_impl()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst/
Dgeneric_indirect.cpp34 const __fp16 *const *const input_ptrs, in a64_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst_indirect_impl()

12