Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_precision_insertion_test.cc55 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
66 return instruction->opcode() == HloOpcode::kCos; in TEST_F()
82 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
93 return instruction->opcode() == HloOpcode::kCos; in TEST_F()
141 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
170 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
186 return instruction->opcode() == HloOpcode::kCos || in TEST_F()
206 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
216 return instruction->opcode() == HloOpcode::kCos; in TEST_F()
235 HloInstruction::CreateUnary(shape, HloOpcode::kCos, a)); in TEST_F()
[all …]
Dhlo_opcode.h78 V(kCos, "cosine", 1) \
Dinstruction_fusion.cc110 case HloOpcode::kCos: in IsExpensive()
Dhlo_instruction.cc695 case HloOpcode::kCos: in CreateUnary()
1401 case HloOpcode::kCos: in CloneWithNewOperands()
1739 case HloOpcode::kCos: in IdenticalSlowPath()
2128 case HloOpcode::kCos: in IsElementwiseImpl()
2587 case HloOpcode::kCos: in Visit()
Dhlo_cost_analysis.cc97 opcode == HloOpcode::kSin || opcode == HloOpcode::kCos) { in HandleElementwiseOp()
Delemental_ir_emitter.cc417 case HloOpcode::kCos: in EmitFloatUnaryOp()
547 case HloOpcode::kCos: { in EmitComplexUnaryOp()
2212 case HloOpcode::kCos: in MakeElementGenerator()
Dhlo_graph_dumper.cc935 case HloOpcode::kCos: in GetInstructionColor()
Dlayout_assignment.cc1995 case HloOpcode::kCos: in InstructionCanChangeLayout()
Dshape_inference.cc252 case HloOpcode::kCos: in InferUnaryOpShape()
Dhlo_evaluator_test.cc352 TestUnaryOp(HloOpcode::kCos, std::move(expected), std::move(operand), in TEST_P()
Dhlo_parser.cc737 case HloOpcode::kCos: in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_precision_test.cc349 [](const HloOpcode opcode) { return opcode == HloOpcode::kCos; }); in XLA_TEST_F()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h48 kCos, enumerator
1195 CosOperator() : Operator(OperatorType::kCos) {}
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc123 CheckSimpleOperator<CosOperator>("COS", OperatorType::kCos); in TEST_F()
Doperator.cc2512 MakeUnique<SimpleOperator<CosOperator>>("COS", OperatorType::kCos)); in BuildOperatorList()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc2108 case OperatorType::kCos: in Run()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3318 return operand.builder()->UnaryOp(HloOpcode::kCos, operand); in Cos()