Home
last modified time | relevance | path

Searched defs:new_dim_size (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Dtensor_layout.cc91 int new_dim_size = new_num_shards_per_dim[dim_ind]; in ExpandShardVector() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorMath.cpp1051 auto new_dim_size = new_buffer.size(dim - 1); in cat_nested_as_jagged() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.cc1846 int64_t dimension, int64_t new_dim_size) { in SplitShardingDimension()
/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py3597 def update_shape(shape, dim, new_dim_size): argument
3725 def update_shape(shape, dim, new_dim_size): argument
3791 def update_shape(shape, dim, new_dim_size): argument