Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_standard.cc121 auto max_dim_size = output_type.getDimSize(dimension); in matchAndRewrite() local
141 int64_t value = (current_value / increase_stride) % max_dim_size; in matchAndRewrite()
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py1088 max_dim_size = max(s.value, minimum_dynamic_dim_size)
1092 max_dim_size = math_ops.maximum(maximum_shapes[idx][i],
1095 max_dim_size = _ceil_to_pow_of_n(max_dim_size, 2)
1097 padding = [0, max_dim_size - input_shape_tensor[i]]