/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | concat_lib_gpu_impl.cu.cc | 40 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()
|
H A D | dynamic_stitch_op_gpu.cu.cc | 35 GpuDeviceArrayStruct<const T*> input_ptrs, in DynamicStitchKernel() 55 const GpuDeviceArrayStruct<const T*>& input_ptrs, in DynamicStitchGPUImpl()
|
H A D | concat_lib_gpu.cc | 41 GpuDeviceArrayOnHost<const T*> input_ptrs(c, inputs_flat.size()); in ConcatGPUCall() local
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst/ |
H A D | generic.cpp | 34 const float *const *const input_ptrs, in sve_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
H A D | benchmark.h | 80 llvm::SmallVector<void*> input_ptrs; member
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const __fp16 *const *const input_ptrs, in a64_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in sme2_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const __fp16 *const *const input_ptrs, in sve_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in sme2_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const __fp16 *const *const input_ptrs, in sve_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in sve_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in sme2_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in sve_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in sve_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_5x5_s1_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in a64_fp32_nhwc_5x5_s1_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in a64_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const __fp16 *const *const input_ptrs, in a64_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in a64_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in sme2_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const __fp16 *const *const input_ptrs, in a64_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const __fp16 *const *const input_ptrs, in a64_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const __fp16 *const *const input_ptrs, in sve_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_5x5_s1_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in sve_fp32_nhwc_5x5_s1_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const float *const *const input_ptrs, in a64_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_indirect_impl()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst/ |
H A D | generic_indirect.cpp | 34 const __fp16 *const *const input_ptrs, in a64_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst_indirect_impl()
|