Searched defs:dims_count (Results 1 – 9 of 9) sorted by relevance
174 const int dims_count = start_indices.size(); in BuildStridedSliceParams() local
459 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local470 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local482 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local495 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local611 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local624 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local636 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local649 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local663 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
443 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local454 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local466 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local479 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local595 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local608 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local620 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local633 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local647 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
94 const int dims_count = output_shape.dimensions_count(); in EvaluateBinaryOperatorOnConstantInputs() local
41 const int dims_count = input_array_a.shape().dimensions_count(); in UnrollBatchMatMul3D() local
736 Py_ssize_t i, dims_count; in append_ast_ext_slice() local
133 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows() local157 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows() local791 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
133 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows() local190 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows() local913 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
45 const int dims_count = in ProcessBroadcastShapes() local3887 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local