Searched refs:rank_x (Results 1 – 2 of 2) sorted by relevance
2001 const int32 rank_x = c->Rank(shape_x); in BroadcastBinaryOpOutputShapeFnHelper() local2003 const int32 rank_out = std::max(rank_x, rank_y); in BroadcastBinaryOpOutputShapeFnHelper()2010 if (rank_x != rank_y) dim_one = c->MakeDim(1); in BroadcastBinaryOpOutputShapeFnHelper()2012 const auto dim_x = i < (rank_out - rank_x) in BroadcastBinaryOpOutputShapeFnHelper()2014 : c->Dim(shape_x, i - (rank_out - rank_x)); in BroadcastBinaryOpOutputShapeFnHelper()
91 int rank_x = input_shape_x.dimensions_count(); in ComputeBinaryOperatorOutputSize() local93 int rank_out = std::max(rank_x, rank_y); in ComputeBinaryOperatorOutputSize()98 int dim_x = i < (rank_out - rank_x) in ComputeBinaryOperatorOutputSize()100 : input_shape_x.dims(i - (rank_out - rank_x)); in ComputeBinaryOperatorOutputSize()