Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc88 bool cur_dim_in_middle = absl::c_linear_search(dims_middle, cur_dim); in PartitionShapeByMiddleDimensions() local
90 if (cur_dim_in_middle) { in PartitionShapeByMiddleDimensions()
94 if (!cur_dim_in_middle) { in PartitionShapeByMiddleDimensions()