Home
last modified time | relevance | path

Searched refs:kCos (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_util.h50 kCos, enumerator
Dtarget_util.cc117 case TargetDeviceFunctionID::kCos: { in GetDeviceFunctionRoot()
Delemental_ir_emitter.cc222 return EmitDeviceMathCall(TargetDeviceFunctionID::kCos, {value}, {prim_type}, in EmitCos()
Dir_emitter_unnested.cc202 case HloOpcode::kCos: in MayPreventVectorization()
219 case HloOpcode::kCos: in MayPreventVectorization()
274 case HloOpcode::kCos: in MayPreventVectorization()
295 case HloOpcode::kCos: in MayPreventVectorization()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_opcode.h79 V(kCos, "cosine", 1) \
Dinstruction_fusion.cc122 case HloOpcode::kCos: in IsExpensive()
Dhlo_instruction.cc895 case HloOpcode::kCos: in CreateUnary()
1722 case HloOpcode::kCos: in CloneWithNewOperands()
2134 case HloOpcode::kCos: in IdenticalSlowPath()
2620 case HloOpcode::kCos: in IsOpElementwise()
3176 case HloOpcode::kCos: in Visit()
Dmemory_space_assignment_test.cc4018 case HloOpcode::kCos: in TEST_P()
4087 case HloOpcode::kCos: in TEST_P()
4234 case HloOpcode::kCos: in TEST_P()
4512 case HloOpcode::kCos: in TEST_P()
4623 case HloOpcode::kCos: in TEST_P()
Dhlo_cost_analysis.cc108 opcode == HloOpcode::kSin || opcode == HloOpcode::kCos || in HandleElementwiseOp()
Delemental_ir_emitter.cc456 case HloOpcode::kCos: in EmitFloatUnaryOp()
593 case HloOpcode::kCos: { in EmitComplexUnaryOp()
2254 case HloOpcode::kCos: in MakeElementGenerator()
Dhlo_graph_dumper.cc950 case HloOpcode::kCos: in GetInstructionColor()
Dsharding_propagation.cc178 case HloOpcode::kCos: in PickRepresentativeOperand()
Dhlo_parser.cc103 case HloOpcode::kCos: in CanInferShape()
1090 case HloOpcode::kCos: in ParseInstructionRhs()
Dlayout_assignment.cc2221 case HloOpcode::kCos: in InstructionCanChangeLayout()
Dshape_inference.cc276 case HloOpcode::kCos: in InferUnaryOpShape()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dop_version.cc241 {{OperatorType::kCos, 1}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
Doperator_test.cc123 CheckSimpleOperator<CosOperator>("COS", OperatorType::kCos); in TEST_F()
Doperator.cc2017 ::tflite::BuiltinOperator_COS, OperatorType::kCos)); in BuildOperatorList()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h49 kCos, enumerator
1249 CosOperator() : Operator(OperatorType::kCos) {}
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc378 return xla::HloOpcode::kCos; in MhloToHloOpcode()
Dhlo_function_importer.cc756 NoAttributeCase(kCos, CosOp); in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc288 case HloOpcode::kCos: in EmitOp()
/external/skia/src/sksl/codegen/
DSkSLVMCodeGenerator.cpp134 kCos, enumerator
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc2191 case OperatorType::kCos: in Run()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3483 case HloOpcode::kCos: in BuildDynamicInferenceGraph()
4764 return operand.builder()->UnaryOp(HloOpcode::kCos, operand); in Cos()

12