Searched defs:accum_depth (Results 1 – 9 of 9) sorted by relevance
44 const int accum_depth = weights_shape.Dims(weights_dims_count - 1); in FullyConnected() local89 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local136 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local189 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected() local
78 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local136 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local225 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
44 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local84 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
78 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local169 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local323 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
460 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local500 int output_stride, int accum_depth, const int32* bias_data, in ShuffledFullyConnectedWorkerImpl()838 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected() local
1666 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local2241 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected() local
114 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in EvalQuantizedInt8() local
51 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
47 const int accum_depth = weights_shape.Dims(weights_dims_count - 1); in FullyConnectedSparseWeight() local