Searched refs:dimension_val (Results 1 – 1 of 1) sorted by relevance
937 int64 dimension_val; in ArgOpShape() local939 dimension_val = dim_t->scalar<int32>()(); in ArgOpShape()941 dimension_val = dim_t->scalar<int64>()(); in ArgOpShape()944 int64 axis = dimension_val < 0 ? dimension_val + input_rank : dimension_val; in ArgOpShape()947 "Dimension (", dimension_val, ") must be in the range [", -input_rank, in ArgOpShape()