Home
last modified time | relevance | path

Searched refs:gemm_input_dims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h2480 const int gemm_input_dims = gemm_input_shape->DimensionsCount(); in Conv() local
2481 int m = FlatSizeSkipDim(*gemm_input_shape, gemm_input_dims - 1); in Conv()
2483 int k = gemm_input_shape->Dims(gemm_input_dims - 1); in Conv()
Doptimized_ops.h1283 const int gemm_input_dims = gemm_input_shape->DimensionsCount(); in Conv() local
1284 int m = FlatSizeSkipDim(*gemm_input_shape, gemm_input_dims - 1); in Conv()
1286 int k = gemm_input_shape->Dims(gemm_input_dims - 1); in Conv()