Home
last modified time | relevance | path

Searched defs:input_multi_index (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc388 std::vector<llvm::Value*> input_multi_index(index.size()); in MakeElementGenerator() local
456 std::vector<llvm::Value*> input_multi_index = in MakeElementGenerator() local
Dir_emitter.cc689 std::vector<llvm::Value*> input_multi_index = in HandleReduce() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc668 std::vector<llvm::Value*> input_multi_index(index.size()); in EmitElementalReduceWindow() local
1058 std::vector<llvm::Value*> input_multi_index(num_dims); in EmitElementalConvolution() local
1595 std::vector<llvm::Value*> input_multi_index = in EmitInnerLoopForVectorizedReduction() local
1838 std::vector<llvm::Value*> input_multi_index = in EmitElementalReduce() local
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1848 std::vector<llvm::Value*> input_multi_index(rank); in EmitElementalDynamicSlice() local