Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc78 std::array<int64, 3> PartitionShapeByMiddleDimensions( in PartitionShapeByMiddleDimensions() function
233 PartitionShapeByMiddleDimensions(input_shape, dims_to_keep); in GetReductionKindAndContiguousComponentsImpl()
246 PartitionShapeByMiddleDimensions(input_shape, dims_to_reduce); in GetReductionKindAndContiguousComponentsImpl()