Home
last modified time | relevance | path

Searched refs:operand_value (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc237 const HloInstruction* op, llvm::Value* operand_value) { in EmitUnaryOp() argument
240 return EmitIntegerUnaryOp(op, operand_value); in EmitUnaryOp()
242 return EmitComplexUnaryOp(op, operand_value); in EmitUnaryOp()
244 return EmitFloatUnaryOp(op, operand_value); in EmitUnaryOp()
249 const HloInstruction* op, llvm::Value* operand_value) { in EmitIntegerUnaryOp() argument
257 return operand_value; in EmitIntegerUnaryOp()
261 ICmpNE(operand_value, in EmitIntegerUnaryOp()
262 llvm::ConstantInt::get(operand_value->getType(), 0)), in EmitIntegerUnaryOp()
266 return IntCast(operand_value, in EmitIntegerUnaryOp()
272 return EmitF32ToBF16(EmitIntegralToFloating(operand_value, from_type, in EmitIntegerUnaryOp()
[all …]
Delemental_ir_emitter.h69 llvm::Value* operand_value);
76 llvm::Value* operand_value);
79 llvm::Value* operand_value);
82 llvm::Value* operand_value);
164 EmitComplexAbsHelper(PrimitiveType prim_type, llvm::Value* operand_value,
168 llvm::Value* operand_value);
171 llvm::Value* operand_value);
173 PrimitiveType prim_type, llvm::Value* operand_value);
177 llvm::Value* operand_value);
181 llvm::Value* operand_value);
[all …]
Dheap_simulator.cc253 for (const HloValue* operand_value : operand_value_set.values()) { in RunComputation() local
255 &alias_analysis.GetBufferContainingValue(*operand_value); in RunComputation()
259 auto it = buffer_live_ranges.find(operand_value); in RunComputation()
273 if (IgnoreBuffer(operand_value)) { in RunComputation()
277 if (!absl::c_linear_search(buffers_freed[i], operand_value)) { in RunComputation()
288 if (value->instruction()->IsUserOf(operand_value->instruction()) && in RunComputation()
291 operand_value->instruction(), operand_value->index(), in RunComputation()
295 Free(operand_value, operand_value->instruction()); in RunComputation()
298 operand_value), in RunComputation()
300 ShareBuffer(value, operand_value, value->instruction()); in RunComputation()
[all …]
Dbuffer_assignment.cc940 const HloValue* operand_value) { in LiveRangeInterferes() argument
941 return user_value->instruction()->IsUserOf(operand_value->instruction()) && in LiveRangeInterferes()
943 operand_value->instruction(), operand_value->index(), in LiveRangeInterferes()
954 auto operand_value = buffer1; in LiveRangeInterferes() local
956 if (!can_share_as_operand(user_value, operand_value)) { in LiveRangeInterferes()
963 auto operand_value = buffer2; in LiveRangeInterferes() local
965 if (!can_share_as_operand(user_value, operand_value)) { in LiveRangeInterferes()
Dhlo_alias_analysis.cc321 const HloValue& operand_value = dataflow_.GetUniqueValueAt( in ComputeInPlaceOperationAliasedBuffers() local
324 VLOG(3) << " operand value " << operand_value.ToShortString() in ComputeInPlaceOperationAliasedBuffers()
326 aliased_buffers->push_back(GetBufferForValue(operand_value)); in ComputeInPlaceOperationAliasedBuffers()