Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h254 std::vector<llvm_ir::ElementGenerator> initial_value_generators,
260 std::vector<llvm_ir::ElementGenerator> initial_value_generators,
Delemental_ir_emitter.cc2500 std::vector<llvm_ir::ElementGenerator> initial_value_generators; in MakeElementGenerator() local
2502 initial_value_generators.push_back(operand_to_generator.at(instr)); in MakeElementGenerator()
2506 std::move(initial_value_generators), index); in MakeElementGenerator()
2516 std::vector<llvm_ir::ElementGenerator> initial_value_generators; in MakeElementGenerator() local
2518 initial_value_generators.push_back(operand_to_generator.at(instr)); in MakeElementGenerator()
2521 std::move(initial_value_generators), index); in MakeElementGenerator()
2592 std::vector<llvm_ir::ElementGenerator> initial_value_generators, in EmitElementalReduceWindow() argument
2619 auto initial_value_generator = initial_value_generators[operand_index]; in EmitElementalReduceWindow()
2711 std::vector<llvm_ir::ElementGenerator> initial_value_generators, in EmitElementalReduce() argument
2739 initial_value_generators[i](llvm_ir::IrArray::Index(index_type))); in EmitElementalReduce()