Searched refs:softmax_op (Results 1 – 9 of 9) sorted by relevance
| /external/XNNPACK/src/ |
| D | softmax-nc.c | 32 xnn_operator_t softmax_op = NULL; in xnn_create_softmax_nc_q8() local
97 softmax_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_softmax_nc_q8()
98 if (softmax_op == NULL) { in xnn_create_softmax_nc_q8()
103 softmax_op->lookup_table = xnn_allocate_simd_memory(256 * sizeof(uint32_t)); in xnn_create_softmax_nc_q8()
104 if (softmax_op->lookup_table == NULL) { in xnn_create_softmax_nc_q8()
109 uint32_t* lookup_table = softmax_op->lookup_table; in xnn_create_softmax_nc_q8()
116 softmax_op->channels = channels; in xnn_create_softmax_nc_q8()
117 softmax_op->input_pixel_stride = input_stride; in xnn_create_softmax_nc_q8()
118 softmax_op->output_pixel_stride = output_stride; in xnn_create_softmax_nc_q8()
120 softmax_op->type = xnn_operator_type_softmax_nc_q8; in xnn_create_softmax_nc_q8()
[all …]
|
| /external/XNNPACK/test/ |
| D | softmax-operator-tester.h | 148 xnn_operator_t softmax_op = nullptr; in TestQ8() local
155 0, &softmax_op)); in TestQ8()
156 ASSERT_NE(nullptr, softmax_op); in TestQ8()
159 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_softmax_op(softmax_op, xnn_dele… in TestQ8()
163 softmax_op, in TestQ8()
169 xnn_run_operator(softmax_op, nullptr /* thread pool */)); in TestQ8()
209 xnn_operator_t softmax_op = nullptr; in TestF32() local
214 0, &softmax_op)); in TestF32()
215 ASSERT_NE(nullptr, softmax_op); in TestF32()
218 …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_q8() local
51 0 /* flags */, &softmax_op); in xnnpack_softmax_q8()
52 if (status != xnn_status_success || softmax_op == nullptr) { in xnnpack_softmax_q8()
58 softmax_op, in xnnpack_softmax_q8()
68 status = xnn_run_operator(softmax_op, nullptr /* thread pool */); in xnnpack_softmax_q8()
75 status = xnn_delete_operator(softmax_op); in xnnpack_softmax_q8()
111 xnn_operator_t softmax_op = nullptr; in xnnpack_softmax_f32() local
114 0 /* flags */, &softmax_op); in xnnpack_softmax_f32()
115 if (status != xnn_status_success || softmax_op == nullptr) { in xnnpack_softmax_f32()
121 softmax_op, in xnnpack_softmax_f32()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | BUILD | 38 "softmax_op.cc",
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | BUILD | 48 "//tensorflow/core/kernels:softmax_op",
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 856 xnn_operator_t softmax_op,
1223 xnn_operator_t softmax_op,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 98 "softmax_op.cc",
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 839 tensorflow::NodeDef* softmax_op = tensorflow_graph->add_node(); in ConvertSoftmaxOperator() local
840 softmax_op->set_op("Softmax"); in ConvertSoftmaxOperator()
841 softmax_op->set_name(src_op.outputs[0]); in ConvertSoftmaxOperator()
842 *softmax_op->add_input() = softmax_input; in ConvertSoftmaxOperator()
845 (*softmax_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertSoftmaxOperator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 4579 ":softmax_op",
4666 name = "softmax_op",
4667 prefix = "softmax_op",
6385 "softmax_op.cc",
|