Searched refs:target_multi_index (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 674 std::vector<llvm::Value*> target_multi_index; in EmitNaiveLlvmIrGemm() local
677 target_multi_index.push_back(lhs_index[dimension]); in EmitNaiveLlvmIrGemm()
682 target_multi_index.push_back(rhs_index[dimension]); in EmitNaiveLlvmIrGemm()
687 target_multi_index, target_array_.GetShape(), lhs_index.GetType()); in EmitNaiveLlvmIrGemm()
|
| D | ir_emitter.cc | 1922 std::vector<llvm::Value*> target_multi_index = in HandleSlice() local
1928 std::replace(target_multi_index.begin(), target_multi_index.end(), in HandleSlice()
1931 llvm_ir::IrArray::Index target_index(target_multi_index, slice->shape(), in HandleSlice()
2481 std::vector<llvm::Value*> target_multi_index = in EmitFastConcatenate() local
2483 std::replace(target_multi_index.begin(), target_multi_index.end(), in EmitFastConcatenate()
2486 llvm_ir::IrArray::Index target_index(target_multi_index, output_shape, in EmitFastConcatenate()
2515 llvm_ir::IrArray::Index source_index(target_multi_index, operand->shape(), in EmitFastConcatenate()
|