Searched refs:operand_ptrs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | mlir_emitter.h | 37 llvm::ArrayRef<llvm::Value *> operand_ptrs, llvm::StringRef func_name,
|
| D | mlir_emitter.cc | 93 llvm::ArrayRef<llvm::Value *> operand_ptrs, llvm::StringRef func_name, in EmitMlirFuncAndCall() argument
136 BuildViewForBuffer(&op_vals, b, operand_shapes[i], operand_ptrs[i]); in EmitMlirFuncAndCall()
|
| D | dot_op_emitter.cc | 256 llvm::Value* operand_ptrs[] = {lhs_array_.GetBasePointer(), in EmitLinalgMatmul() local
272 operand_ptrs, name, [&](mlir::OpBuilder* builder, mlir::FuncOp function) { in EmitLinalgMatmul()
|
| D | ir_emitter.cc | 1082 std::vector<llvm::Value*> operand_ptrs; in HandleAllReduceSingleReplica() local
1091 operand_ptrs.push_back(EmitBufferPointer(out_slice, operand_shape)); in HandleAllReduceSingleReplica()
1094 MemCpy(operand_ptrs.back(), /*DstAlign=*/llvm::Align(1), in_ptr, in HandleAllReduceSingleReplica()
1097 llvm_ir::EmitTuple(GetIrArrayFor(crs), operand_ptrs, &b_); in HandleAllReduceSingleReplica()
|