Home
last modified time | relevance | path

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

12

/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.cc2186 case OperatorType::kFloor: in Run()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_opcode.h89 V(kFloor, "floor", 1) \
Dinstruction_fusion_test.cc320 HloInstruction::CreateUnary(shape, HloOpcode::kFloor, param0)); in TEST_F()
Dinstruction_fusion.cc85 case HloOpcode::kFloor: in IsExpensive()
Dcall_graph_test.cc253 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kFloor)); in TEST_F()
Dhlo_instruction.cc899 case HloOpcode::kFloor: in CreateUnary()
1727 case HloOpcode::kFloor: in CloneWithNewOperands()
2139 case HloOpcode::kFloor: in IdenticalSlowPath()
2623 case HloOpcode::kFloor: in IsOpElementwise()
3164 case HloOpcode::kFloor: in Visit()
Dhlo_graph_dumper.cc954 case HloOpcode::kFloor: in GetInstructionColor()
Dsharding_propagation.cc186 case HloOpcode::kFloor: in PickRepresentativeOperand()
Delemental_ir_emitter.cc468 case HloOpcode::kFloor: in EmitFloatUnaryOp()
2257 case HloOpcode::kFloor: in MakeElementGenerator()
Dhlo_parser.cc110 case HloOpcode::kFloor: in CanInferShape()
1095 case HloOpcode::kFloor: in ParseInstructionRhs()
Dlayout_assignment.cc2231 case HloOpcode::kFloor: in InstructionCanChangeLayout()
Dshape_inference.cc266 case HloOpcode::kFloor: in InferUnaryOpShape()
Dbuffer_assignment_test.cc1092 BuildR0F32UnaryOpComputation(HloOpcode::kFloor, "Floor")); in TEST_F()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dop_version.cc230 {{OperatorType::kFloor, 1}, "1.9.0"}, in GetMinimumRuntimeVersionForModel()
Doperator_test.cc113 CheckSimpleOperator<FloorOperator>("FLOOR", OperatorType::kFloor); in TEST_F()
Doperator.cc1995 ::tflite::BuiltinOperator_FLOOR, OperatorType::kFloor)); in BuildOperatorList()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h86 kFloor, enumerator
1758 FloorOperator() : Operator(OperatorType::kFloor) {}
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc384 return xla::HloOpcode::kFloor; in MhloToHloOpcode()
Dhlo_function_importer.cc760 NoAttributeCase(kFloor, FloorOp); in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc304 case HloOpcode::kFloor: in EmitOp()
/external/skia/src/sksl/codegen/
DSkSLVMCodeGenerator.cpp154 kFloor, enumerator
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3487 case HloOpcode::kFloor: in BuildDynamicInferenceGraph()
4740 return operand.builder()->UnaryOp(HloOpcode::kFloor, operand); in Floor()

12