Home
last modified time | relevance | path

Searched refs:kFloor (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dfloor_ceil_test.cc35 kFloor, enumerator
48 ASSERT_EQ(kFloor, f); in TestR1F32()
61 ASSERT_EQ(kFloor, f); in TestR0F32()
78 XLA_TEST_F(FloorCeilTest, R1S0Floor) { TestR1F32({}, {}, kFloor); } in XLA_TEST_F()
82 {0.0, -0.0, infinity_, minus_infinity_, 1.0, -1.0}, kFloor); in TEST_F()
91 TestR0F32(0.0, 0.0, kFloor); in TEST_F()
92 TestR0F32(-0.0, -0.0, kFloor); in TEST_F()
93 TestR0F32(infinity_, infinity_, kFloor); in TEST_F()
94 TestR0F32(minus_infinity_, minus_infinity_, kFloor); in TEST_F()
95 TestR0F32(1.1, 1.0, kFloor); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc105 HloInstruction::CreateUnary(vshape, HloOpcode::kFloor, exp)); in TEST_F()
181 HloInstruction::CreateUnary(cshape, HloOpcode::kFloor, exp)); in TEST_F()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dreorder_elementwise_unary.cc35 case OperatorType::kFloor: in IsElementwiseOperator()
Dpropagate_fixed_sizes.cc2104 case OperatorType::kFloor: in Run()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_opcode.h88 V(kFloor, "floor", 1) \
Dinstruction_fusion_test.cc358 HloInstruction::CreateUnary(shape, HloOpcode::kFloor, param0)); in TEST_F()
Dinstruction_fusion.cc76 case HloOpcode::kFloor: in IsExpensive()
Dhlo_instruction.cc699 case HloOpcode::kFloor: in CreateUnary()
1406 case HloOpcode::kFloor: in CloneWithNewOperands()
1744 case HloOpcode::kFloor: in IdenticalSlowPath()
2131 case HloOpcode::kFloor: in IsElementwiseImpl()
2575 case HloOpcode::kFloor: in Visit()
Dcall_graph_test.cc253 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kFloor)); in TEST_F()
Dhlo_graph_dumper.cc939 case HloOpcode::kFloor: in GetInstructionColor()
Delemental_ir_emitter.cc427 case HloOpcode::kFloor: in EmitFloatUnaryOp()
2215 case HloOpcode::kFloor: in MakeElementGenerator()
Dlayout_assignment.cc2005 case HloOpcode::kFloor: in InstructionCanChangeLayout()
Dshape_inference.cc242 case HloOpcode::kFloor: in InferUnaryOpShape()
Dbuffer_assignment_test.cc920 BuildR0F32UnaryOpComputation(HloOpcode::kFloor, "Floor")); in TEST_F()
Dhlo_dataflow_analysis_test.cc1803 CreateR0F32UnaryOpComputation(HloOpcode::kFloor)); in TEST_P()
Dhlo_parser.cc742 case HloOpcode::kFloor: in ParseInstructionRhs()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h84 kFloor, enumerator
1704 FloorOperator() : Operator(OperatorType::kFloor) {}
Dexport_tensorflow.cc2207 } else if (src_op.type == OperatorType::kFloor) { in ConvertOperator()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc114 CheckSimpleOperator<FloorOperator>("FLOOR", OperatorType::kFloor); in TEST_F()
Doperator.cc2495 MakeUnique<SimpleOperator<FloorOperator>>("FLOOR", OperatorType::kFloor)); in BuildOperatorList()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3297 return operand.builder()->UnaryOp(HloOpcode::kFloor, operand); in Floor()