Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h32 float* im2col_data) { in Conv()
115 uint8_t* im2col_data, void* cpu_backend_context) { in Conv()
209 const RuntimeShape& im2col_shape, int8_t* im2col_data, in HybridConvPerChannel()
Dtranspose_conv.h32 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv()
122 uint8_t* im2col_data, int32_t* scratch_buffer) { in TransposeConv()
Dlegacy_reference_ops.h215 float* im2col_data, const Dims<4>& im2col_dims) { in Conv()
239 float* output_data, const Dims<4>& output_dims, float* im2col_data, in Conv()
256 const Dims<4>& output_dims, float* im2col_data, in Conv()
272 const Dims<4>& output_dims, float* im2col_data, in Conv()
288 uint8* im2col_data, const Dims<4>& im2col_dims, in Conv()
321 const Dims<4>& output_dims, uint8* im2col_data, in Conv()
341 const Dims<4>& output_dims, uint8* im2col_data, in Conv()
369 uint8* output_data, const Dims<4>& output_dims, uint8* im2col_data, in Conv()
382 const Dims<4>& output_dims, float* im2col_data, in TransposeConv()
403 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dtranspose_conv.h32 int8_t* output_data, const RuntimeShape& im2col_shape, int8_t* im2col_data, in TransposeConv()
132 int16_t* output_data, const RuntimeShape& im2col_shape, int8_t* im2col_data, in TransposeConv()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dim2col_utils.h121 const RuntimeShape& output_shape, T* im2col_data, in DilatedIm2col()
207 const RuntimeShape& output_shape, T* im2col_data) { in DilatedIm2col()
377 T* im2col_data) { in Im2col3D()
422 const RuntimeShape& im2col_shape, T* im2col_data) { in DilatedIm2col3D()
Dmultithreaded_conv.h141 float* im2col_data) { in Conv()
Dlegacy_optimized_ops.h2379 T* im2col_data) { in DilatedIm2col()
2428 float* im2col_data) { in Conv()
2533 float* im2col_data, const Dims<4>& im2col_dims) { in Conv()
2559 int8_t* im2col_data, const Dims<4>& im2col_dims, in HybridConv()
2584 float* output_data, const Dims<4>& output_dims, float* im2col_data, in Conv()
2601 const Dims<4>& output_dims, float* im2col_data, in Conv()
2617 const Dims<4>& output_dims, float* im2col_data, in Conv()
2629 uint8* im2col_data, gemmlowp::GemmContext* gemmlowp_context) { in Conv()
2739 uint8* im2col_data, const Dims<4>& im2col_dims, in Conv()
2772 const Dims<4>& output_dims, uint8* im2col_data, in Conv()
[all …]
Doptimized_ops.h912 float* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv()
1011 int8_t* im2col_data, CpuBackendContext* context) { in HybridConv()
1100 const RuntimeShape& im2col_shape, int8_t* im2col_data, in HybridConvPerChannel()
1210 uint8* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv()
4824 const RuntimeShape& output_shape, T* im2col_data) { in TransposeIm2col()
8052 float* im2col_data, in Conv3D()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dconv.h39 InputScalar* im2col_data, CpuBackendContext* cpu_backend_context) { in ConvPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dconv_per_channel_quantized_16x8_test.cc285 float im2col_data; in TryTestOneConvFilter() local