Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc646 const auto load_memrefs = [loc, &b](mlir::ValueRange range) { in EmitUsingElementalIrEmitter() local
666 std::vector<mlir::Value> operands = load_memrefs(reduce.operands()); in EmitUsingElementalIrEmitter()
667 std::vector<mlir::Value> init_values = load_memrefs(reduce.init_values()); in EmitUsingElementalIrEmitter()
701 std::vector<mlir::Value> loads = load_memrefs(operands); in EmitUsingElementalIrEmitter()