Searched refs:dimension_val (Results 1 – 3 of 3) sorted by relevance
130 T dimension_val = (perm[i] < dimensions.num_dimensions()) ? dimensions_copy[perm[i]] : 0; in permute() local131 dimensions.set(i, dimension_val); in permute()147 size_t dimension_val = (perm[i] < shape.num_dimensions()) ? shape_copy[perm[i]] : 1; in permute() local148 shape.set(i, dimension_val, false, false); // Avoid changes in _num_dimension in permute()
733 T dimension_val = old_dim[i]; in permute_strides() local734 dimensions.set(perm[i], dimension_val); in permute_strides()
1107 int64_t dimension_val; in ArgOpShape() local1109 dimension_val = dim_t->scalar<int32>()(); in ArgOpShape()1111 dimension_val = dim_t->scalar<int64_t>()(); in ArgOpShape()1114 int64_t axis = dimension_val < 0 ? dimension_val + input_rank : dimension_val; in ArgOpShape()1117 "Dimension (", dimension_val, ") must be in the range [", -input_rank, in ArgOpShape()