Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h587 mlir::Operation* unnested_hlo, const Shape& reduction_operand_shape,
Dir_emitter_unnested.cc4595 mlir::Operation* unnested_hlo, const Shape& reduction_operand_shape, in EmitTileElementForReduction() argument
4633 GetUnnormalizedIndex(index, reduction_operand_shape, &b_, in EmitTileElementForReduction()
4641 input_index.multidim(), reduction_operand_shape, input_index.GetType()); in EmitTileElementForReduction()