Searched refs:SOFTMAX (Results 1 – 22 of 22) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | softmax_test.cc | 49 SingleOpModel model({ToString(OperationType::SOFTMAX), attr}, {input}, in TEST()
71 SingleOpModel model({ToString(OperationType::SOFTMAX), attr}, {input}, in TEST()
91 SingleOpModel model({ToString(OperationType::SOFTMAX), attr}, {input}, in TEST()
114 SingleOpModel model({ToString(OperationType::SOFTMAX), attr}, {input}, in TEST()
145 SingleOpModel model({ToString(OperationType::SOFTMAX), attr}, {input}, in TEST()
180 SingleOpModel model({ToString(OperationType::SOFTMAX), attr}, {input}, in TEST()
|
| D | registry.cc | 96 insert_op(Type::SOFTMAX, NewSoftmaxNodeShader); in Registry()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | quantization.mlir | 21 // CHECK-NEXT: builtin_code: SOFTMAX
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operations.h | 85 SOFTMAX, enumerator
|
| D | operations.cc | 173 case OperationType::SOFTMAX: in ToString()
249 {"softmax", OperationType::SOFTMAX}, in OperationTypeFromString()
|
| D | model_builder.cc | 1764 node->operation.type = ToString(OperationType::SOFTMAX); in Parse()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v0.fbs | 73 SOFTMAX = 17,
|
| D | schema_v1.fbs | 78 SOFTMAX = 17,
|
| D | schema_v2.fbs | 87 SOFTMAX = 25,
|
| D | schema_v3.fbs | 97 SOFTMAX = 25,
|
| D | schema_v3a.fbs | 249 SOFTMAX = 25,
|
| D | schema.fbs | 253 SOFTMAX = 25,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | README.md | 47 * `SOFTMAX v1`
|
| /external/tensorflow/tensorflow/lite/tools/benchmark/ |
| D | README.md | 286 …SOFTMAX 79.443 0.029 0.029 0.001% 1.993% 0.000 0 [MobilenetV1/Predicti…
306 SOFTMAX 1 0.000 0.000% 100.000% 0.000 0
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | operation_selector.cc | 484 case OperationType::SOFTMAX: { in GPUOperationFromNode()
|
| /external/tensorflow/tensorflow/lite/g3doc/performance/ |
| D | quantization_spec.md | 275 SOFTMAX
|
| D | gpu_advanced.md | 57 * `SOFTMAX`
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | README.md | 304 ### `SOFTMAX`
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | optimize.mlir | 1651 …// CHECK: %[[SOFTMAX:.*]] = "tfl.softmax"(%arg0) {beta = -1.000000e+00 : f32} : (tensor<16x1024xf3…
1652 …// CHECK: %[[ARG_MAX:.*]] = "tfl.arg_max"(%[[SOFTMAX]], %[[CST]]) : (tensor<16x1024xf32>, tensor<1…
1663 …// CHECK: %[[SOFTMAX:.*]] = "tfl.softmax"(%arg0) {beta = 1.000000e+00 : f32} : (tensor<16x1024xf32…
1664 …// CHECK: %[[ARG_MAX:.*]] = "tfl.arg_max"(%[[SOFTMAX]], %[[CST]]) : (tensor<16x1024xf32>, tensor<1…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | lower_tf.mlir | 416 // CHECK-DAG: %[[SOFTMAX:.*]] = "tf.Softmax"(%[[FEATURES]]) : (tensor<2x3xf32>) -> tensor<2x3xf32>
417 …// CHECK-DAG: %[[BACKPROP:.*]] = "tf.Sub"(%[[SOFTMAX]], %[[LABELS]]) : (tensor<2x3xf32>, tensor<2x…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | test_schema.fbs | 246 SOFTMAX = 25,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 106 ADD_LAYER(SOFTMAX) in LayerTypeToString()
|