Home
last modified time | relevance | path

Searched refs: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
393 input_multi_index[i] = NSWSub( in MakeElementGenerator()
402 SRem(input_multi_index[i], in MakeElementGenerator()
408 input_multi_index[i] = in MakeElementGenerator()
409 SDiv(input_multi_index[i], in MakeElementGenerator()
419 ICmpULT(input_multi_index[i], in MakeElementGenerator()
428 IrArray::Index input_index(input_multi_index, operand->shape(), in MakeElementGenerator()
456 std::vector<llvm::Value*> input_multi_index = in MakeElementGenerator() local
463 for (auto& i : input_multi_index) { in MakeElementGenerator()
471 input_multi_index, hlo->operand(0)->shape(), index_type); in MakeElementGenerator()
Dir_emitter.cc689 std::vector<llvm::Value*> input_multi_index = in HandleReduce() local
702 for (auto& i : input_multi_index) { in HandleReduce()
710 llvm_ir::IrArray::Index input_index(input_multi_index, arg->shape(), in HandleReduce()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc668 std::vector<llvm::Value*> input_multi_index(index.size()); in EmitElementalReduceWindow() local
673 input_multi_index[i] = NSWSub( in EmitElementalReduceWindow()
681 ICmpEQ(SRem(input_multi_index[i], in EmitElementalReduceWindow()
691 input_multi_index[i] = in EmitElementalReduceWindow()
692 SDiv(input_multi_index[i], in EmitElementalReduceWindow()
700 ICmpULT(input_multi_index[i], in EmitElementalReduceWindow()
715 llvm_ir::IrArray::Index input_index(input_multi_index, operand->shape(), in EmitElementalReduceWindow()
1058 std::vector<llvm::Value*> input_multi_index(num_dims); in EmitElementalConvolution() local
1060 input_multi_index[dnums.input_spatial_dimensions(i)] = input_spatial[i]; in EmitElementalConvolution()
1062 input_multi_index[dnums.input_feature_dimension()] = input_feature; in EmitElementalConvolution()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1848 std::vector<llvm::Value*> input_multi_index(rank); in EmitElementalDynamicSlice() local
1852 input_multi_index[i] = Add(slice_start_multi_index[i], index[i]); in EmitElementalDynamicSlice()
1854 llvm_ir::IrArray::Index input_index(input_multi_index, input_hlo->shape(), in EmitElementalDynamicSlice()