Searched refs:softmax_op (Results 1 – 9 of 9) sorted by relevance
| /external/XNNPACK/src/operators/ |
| D | softmax-nc.c | 32 xnn_operator_t softmax_op = NULL; in xnn_create_softmax_nc_qu8() local
98 softmax_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_softmax_nc_qu8()
99 if (softmax_op == NULL) { in xnn_create_softmax_nc_qu8()
106 softmax_op->lookup_table = xnn_allocate_simd_memory(256 * sizeof(uint32_t)); in xnn_create_softmax_nc_qu8()
107 if (softmax_op->lookup_table == NULL) { in xnn_create_softmax_nc_qu8()
114 uint32_t* lookup_table = softmax_op->lookup_table; in xnn_create_softmax_nc_qu8()
121 softmax_op->channels = channels; in xnn_create_softmax_nc_qu8()
122 softmax_op->input_pixel_stride = input_stride; in xnn_create_softmax_nc_qu8()
123 softmax_op->output_pixel_stride = output_stride; in xnn_create_softmax_nc_qu8()
125 softmax_op->type = xnn_operator_type_softmax_nc_qu8; in xnn_create_softmax_nc_qu8()
[all …]
|
| /external/XNNPACK/test/ |
| D | softmax-operator-tester.h | 149 xnn_operator_t softmax_op = nullptr; in TestQU8() local
156 0, &softmax_op)); in TestQU8()
157 ASSERT_NE(nullptr, softmax_op); in TestQU8()
160 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_softmax_op(softmax_op, xnn_dele… in TestQU8()
164 softmax_op, in TestQU8()
170 xnn_run_operator(softmax_op, nullptr /* thread pool */)); in TestQU8()
210 xnn_operator_t softmax_op = nullptr; in TestF32() local
215 0, &softmax_op)); in TestF32()
216 ASSERT_NE(nullptr, softmax_op); in TestF32()
219 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_softmax_op(softmax_op, xnn_dele… in TestF32()
[all …]
|
| /external/XNNPACK/bench/ |
| D | softmax.cc | 46 xnn_operator_t softmax_op = nullptr; in xnnpack_softmax_qu8() local
51 0 /* flags */, &softmax_op); in xnnpack_softmax_qu8()
52 if (status != xnn_status_success || softmax_op == nullptr) { in xnnpack_softmax_qu8()
58 softmax_op, in xnnpack_softmax_qu8()
68 status = xnn_run_operator(softmax_op, nullptr /* thread pool */); in xnnpack_softmax_qu8()
75 status = xnn_delete_operator(softmax_op); in xnnpack_softmax_qu8()
114 xnn_operator_t softmax_op = nullptr; in xnnpack_softmax_f32() local
117 0 /* flags */, &softmax_op); in xnnpack_softmax_f32()
118 if (status != xnn_status_success || softmax_op == nullptr) { in xnnpack_softmax_f32()
124 softmax_op, in xnnpack_softmax_f32()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | BUILD | 39 "softmax_op.cc",
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | BUILD | 49 "//tensorflow/core/kernels:softmax_op",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 103 "softmax_op.cc",
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 1565 xnn_operator_t softmax_op,
2197 xnn_operator_t softmax_op,
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 845 tensorflow::NodeDef* softmax_op = tensorflow_graph->add_node(); in ConvertSoftmaxOperator() local
846 softmax_op->set_op("Softmax"); in ConvertSoftmaxOperator()
847 softmax_op->set_name(src_op.outputs[0]); in ConvertSoftmaxOperator()
848 *softmax_op->add_input() = softmax_input; in ConvertSoftmaxOperator()
851 (*softmax_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertSoftmaxOperator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3893 ":softmax_op",
3980 name = "softmax_op",
3981 prefix = "softmax_op",
5849 "softmax_op.cc",
|