Searched refs:target_multi_index (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 609 std::vector<llvm::Value*> target_multi_index; in HandleDot() local
612 target_multi_index.push_back(lhs_index[dimension]); in HandleDot()
619 target_multi_index.push_back(rhs_index[dimension]); in HandleDot()
623 llvm_ir::IrArray::Index target_index(target_multi_index, in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 525 std::vector<llvm::Value*> target_multi_index; in EmitNaiveLlvmIrGemm() local
528 target_multi_index.push_back(lhs_index[dimension]); in EmitNaiveLlvmIrGemm()
533 target_multi_index.push_back(rhs_index[dimension]); in EmitNaiveLlvmIrGemm()
538 target_multi_index, target_array_.GetShape(), lhs_index.GetType()); in EmitNaiveLlvmIrGemm()
|
| D | ir_emitter.cc | 2008 std::vector<llvm::Value*> target_multi_index = in HandleSlice() local
2014 std::replace(target_multi_index.begin(), target_multi_index.end(), in HandleSlice()
2017 llvm_ir::IrArray::Index target_index(target_multi_index, slice->shape(), in HandleSlice()
2420 std::vector<llvm::Value*> target_multi_index = in EmitFastConcatenate() local
2422 std::replace(target_multi_index.begin(), target_multi_index.end(), in EmitFastConcatenate()
2425 llvm_ir::IrArray::Index target_index(target_multi_index, output_shape, in EmitFastConcatenate()
|