Searched defs:dims_count (Results 1 – 11 of 11) sorted by relevance
30 const int dims_count = output_shape.DimensionsCount(); in Densify() local
41 const int dims_count = in ProcessBroadcastShapes() local
44 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local
174 const int dims_count = start_indices.size(); in BuildStridedSliceParams() local
484 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local495 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local507 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local520 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local636 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local649 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local661 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local674 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local688 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
95 const int dims_count = output_shape.dimensions_count(); in EvaluateBinaryOperatorOnConstantInputs() local
74 int dims_count = in GetDeviceWorkDimsSizes() local
612 const int dims_count = shape.size(); in AddSparseTensor() local
743 Py_ssize_t i, dims_count; in append_ast_ext_slice() local
144 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows() local168 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows() local281 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
659 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local