Searched defs:accum_depth (Results 1 – 7 of 7) 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
79 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local137 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local226 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
458 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local498 int output_stride, int accum_depth, const int32* bias_data, in ShuffledFullyConnectedWorkerImpl()836 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected() local
1665 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local2240 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected() local
47 const int accum_depth = weights_shape.Dims(weights_dims_count - 1); in FullyConnectedSparseWeight() local