Home
last modified time | relevance | path

Searched defs:accum_depth (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dfully_connected.h51 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnectedPerChannel() local
92 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnectedPerChannel() local
135 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
177 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dfully_connected.h47 const int accum_depth = weights_shape.Dims(weights_dims_count - 1); in FullyConnected() local
92 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
139 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
192 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected() local
H A Dbatch_matmul.h79 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
137 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
226 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dbatch_matmul.h78 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
169 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
323 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
429 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
H A Doptimized_ops.h396 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
436 int output_stride, int accum_depth, const int32* bias_data, in ShuffledFullyConnectedWorkerImpl()
774 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected() local
H A Dlegacy_optimized_ops.h1667 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
2242 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Duniform_quantized_dot_ops.cc63 const int64_t accum_depth = rhs.dim_size(0); in DotWithAccFunctionAndOutputFunction() local
138 const int64_t accum_depth = lhs.dim_size(1); in EvalHybridDot() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
H A Dfully_connected.h50 const int accum_depth = weights_shape.Dims(weights_dims_count - 1); in FullyConnectedSparseWeight() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfully_connected.cc923 const int accum_depth = weights_shape.Dims(weights_dims_count - 1); in VerifySparsity() local