Searched refs:floor_op (Results 1 – 6 of 6) sorted by relevance
| /external/XNNPACK/test/ |
| D | floor-operator-tester.h | 105 xnn_operator_t floor_op = nullptr; in TestF32() local
110 0, &floor_op)); in TestF32()
111 ASSERT_NE(nullptr, floor_op); in TestF32()
114 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_floor_op(floor_op, xnn_delete_o… in TestF32()
118 floor_op, in TestF32()
124 xnn_run_operator(floor_op, nullptr /* thread pool */)); in TestF32()
|
| /external/XNNPACK/bench/ |
| D | floor.cc | 47 xnn_operator_t floor_op = nullptr; in xnnpack_floor_f32() local
50 0 /* flags */, &floor_op); in xnnpack_floor_f32()
51 if (status != xnn_status_success || floor_op == nullptr) { in xnnpack_floor_f32()
57 floor_op, in xnnpack_floor_f32()
67 status = xnn_run_operator(floor_op, nullptr /* thread pool */); in xnnpack_floor_f32()
74 status = xnn_delete_operator(floor_op); in xnnpack_floor_f32()
|
| /external/XNNPACK/src/operators/ |
| D | unary-elementwise-nc.c | 623 xnn_operator_t floor_op, in xnn_setup_floor_nc_f32() argument
629 if (floor_op->type != xnn_operator_type_floor_nc_f32) { in xnn_setup_floor_nc_f32()
632 xnn_operator_type_to_string(floor_op->type)); in xnn_setup_floor_nc_f32()
635 floor_op->state = xnn_run_state_invalid; in xnn_setup_floor_nc_f32()
638 floor_op, in xnn_setup_floor_nc_f32()
641 &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 | 1203 tensorflow::NodeDef* floor_op = tensorflow_graph->add_node(); in ConvertFloorOperator() local
1204 floor_op->set_op("Floor"); in ConvertFloorOperator()
1205 floor_op->set_name(src_op.outputs[0]); in ConvertFloorOperator()
1207 *floor_op->add_input() = src_op.inputs[0]; in ConvertFloorOperator()
1208 (*floor_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertFloorOperator()
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 1351 xnn_operator_t floor_op,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1324 auto floor_op = in matchAndRewrite() local
1326 rewriter.replaceOp(op, floor_op.getResult()); in matchAndRewrite()
|