Home
last modified time | relevance | path

Searched refs:kExpm1 (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_util.h53 kExpm1, enumerator
Dtarget_util.cc126 case TargetDeviceFunctionID::kExpm1: { in GetDeviceFunctionRoot()
Delemental_ir_emitter.cc234 return EmitDeviceMathCall(TargetDeviceFunctionID::kExpm1, {value}, in EmitExpm1()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_opcode.h87 V(kExpm1, "exponential-minus-one", 1) \
Dinstruction_fusion.cc160 case HloOpcode::kExpm1: in IsExpensive()
Dhlo_instruction.cc898 case HloOpcode::kExpm1: in CreateUnary()
1724 case HloOpcode::kExpm1: in CloneWithNewOperands()
2138 case HloOpcode::kExpm1: in IdenticalSlowPath()
2622 case HloOpcode::kExpm1: in IsOpElementwise()
3162 case HloOpcode::kExpm1: in Visit()
Dhlo_cost_analysis.cc109 opcode == HloOpcode::kExpm1 || opcode == HloOpcode::kLog1p || in HandleElementwiseOp()
Delemental_ir_emitter.cc450 case HloOpcode::kExpm1: in EmitFloatUnaryOp()
579 case HloOpcode::kExpm1: { in EmitComplexUnaryOp()
2256 case HloOpcode::kExpm1: in MakeElementGenerator()
Dhlo_graph_dumper.cc953 case HloOpcode::kExpm1: in GetInstructionColor()
Dsharding_propagation.cc185 case HloOpcode::kExpm1: in PickRepresentativeOperand()
Dhlo_parser.cc108 case HloOpcode::kExpm1: in CanInferShape()
1092 case HloOpcode::kExpm1: in ParseInstructionRhs()
Dlayout_assignment.cc2229 case HloOpcode::kExpm1: in InstructionCanChangeLayout()
Dshape_inference.cc279 case HloOpcode::kExpm1: in InferUnaryOpShape()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc382 return xla::HloOpcode::kExpm1; in MhloToHloOpcode()
Dhlo_function_importer.cc759 NoAttributeCase(kExpm1, Expm1Op); in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc302 case HloOpcode::kExpm1: in EmitOp()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3486 case HloOpcode::kExpm1: in BuildDynamicInferenceGraph()
4737 return operand.builder()->UnaryOp(HloOpcode::kExpm1, operand); in Expm1()