Home
last modified time | relevance | path

Searched defs:output_dims_count (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h42 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnectedSparseWeight() local
89 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnectedSparseWeight1x4Impl() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_unary.cc211 const int output_dims_count = output_shape.dimensions_count(); in Run() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h39 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnected() local
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected.cc517 const int output_dims_count = output_shape.DimensionsCount(); in EvalSparseHybridImpl() local