Searched refs:floor_op (Results 1 – 6 of 6) sorted by relevance
| /external/XNNPACK/test/ |
| D | floor-operator-tester.h | 106 xnn_operator_t floor_op = nullptr; in TestF16() local
110 0, &floor_op); in TestF16()
115 ASSERT_NE(nullptr, floor_op); in TestF16()
118 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_floor_op(floor_op, xnn_delete_o… in TestF16()
122 floor_op, in TestF16()
128 xnn_run_operator(floor_op, nullptr /* thread pool */)); in TestF16()
162 xnn_operator_t floor_op = nullptr; in TestF32() local
167 0, &floor_op)); in TestF32()
168 ASSERT_NE(nullptr, floor_op); in TestF32()
171 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_floor_op(floor_op, xnn_delete_o… in TestF32()
[all …]
|
| /external/XNNPACK/bench/ |
| D | floor.cc | 46 xnn_operator_t floor_op = nullptr; in xnnpack_floor_f32() local
49 0 /* flags */, &floor_op); in xnnpack_floor_f32()
50 if (status != xnn_status_success || floor_op == nullptr) { in xnnpack_floor_f32()
56 floor_op, batch_size, in xnnpack_floor_f32()
65 status = xnn_run_operator(floor_op, nullptr /* thread pool */); in xnnpack_floor_f32()
72 status = xnn_delete_operator(floor_op); in xnnpack_floor_f32()
|
| /external/XNNPACK/src/operators/ |
| D | unary-elementwise-nc.c | 1622 xnn_operator_t floor_op, in xnn_setup_floor_nc_f16() argument
1629 floor_op, xnn_operator_type_floor_nc_f16, in xnn_setup_floor_nc_f16()
1638 xnn_operator_t floor_op, in xnn_setup_floor_nc_f32() argument
1645 floor_op, xnn_operator_type_floor_nc_f32, in xnn_setup_floor_nc_f32()
1649 &floor_op->params.f32_rnd, sizeof(floor_op->params.f32_rnd), in xnn_setup_floor_nc_f32()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1208 tensorflow::NodeDef* floor_op = tensorflow_graph->add_node(); in ConvertFloorOperator() local
1209 floor_op->set_op("Floor"); in ConvertFloorOperator()
1210 floor_op->set_name(src_op.outputs[0]); in ConvertFloorOperator()
1212 *floor_op->add_input() = src_op.inputs[0]; in ConvertFloorOperator()
1213 (*floor_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertFloorOperator()
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 1779 xnn_operator_t floor_op,
2503 xnn_operator_t floor_op,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1606 auto floor_op = in matchAndRewrite() local
1608 rewriter.replaceOp(op, floor_op.getResult()); in matchAndRewrite()
|