Home
last modified time | relevance | path

Searched defs:max_dim_size (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dsharding_util_ops.cc420 const int max_dim_size = slice_shape.dim_size(dim) * num_concats_[dim]; in GetInputsAndOutputShape() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dsharding_util_ops.cc708 const int max_dim_size = slice_shape.dim_size(i) * num_concats_[i]; in GetInputsAndOutputShape() local