| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | softmax.h | 29 class Softmax : public GPUOperation {
31 Softmax() = default;
32 explicit Softmax(const OperationDef& definition) : GPUOperation(definition) {} in Softmax() function
39 Softmax(Softmax&& kernel);
40 Softmax& operator=(Softmax&& kernel);
41 Softmax(const Softmax&) = delete;
42 Softmax& operator=(const Softmax&) = delete;
44 friend Softmax CreateSoftmax();
53 Softmax CreateSoftmax(const OperationDef& definition);
|
| D | softmax.cc | 68 Softmax::Softmax(Softmax&& kernel) in Softmax() function in tflite::gpu::cl::Softmax
73 Softmax& Softmax::operator=(Softmax&& kernel) { in operator =()
82 Status Softmax::Compile(const CreationContext& creation_context) { in Compile()
89 Status Softmax::BindArguments() { in BindArguments()
100 int3 Softmax::GetGridSize() const { in GetGridSize()
107 Status Softmax::Tune(const TuningParameters& params) { in Tune()
112 Status Softmax::AddToQueue(CLCommandQueue* queue) { in AddToQueue()
117 Softmax CreateSoftmax(const OperationDef& definition) { in CreateSoftmax()
118 return Softmax(definition); in CreateSoftmax()
|
| D | softmax_test.cc | 36 TEST_F(OpenCLOperationTest, Softmax) { in TEST_F() argument
51 Softmax operation = CreateSoftmax(op_def); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | softmax.cc | 87 tflite::reference_ops::Softmax(input->data.f, input_size, 1, params->beta, in Softmax1DFloat()
96 tflite::reference_ops::Softmax(input->data.f, input_size, batch_size, in Softmax2DFloat()
114 tflite::reference_ops::Softmax(op_params, shape, in Softmax1DQuantized()
119 tflite::reference_integer_ops::Softmax( in Softmax1DQuantized()
123 tflite::reference_integer_ops::Softmax( in Softmax1DQuantized()
145 tflite::reference_ops::Softmax(op_params, shape, in Softmax2DQuantized()
150 tflite::reference_integer_ops::Softmax( in Softmax2DQuantized()
154 tflite::reference_integer_ops::Softmax( in Softmax2DQuantized()
166 tflite::reference_ops::Softmax( in Softmax4DFloat()
178 tflite::reference_ops::Softmax( in Softmax4DQuantized()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_Softmax.pbtxt | 2 graph_op_name: "Softmax"
4 name: "nn.Softmax"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Softmax.pbtxt | 2 name: "Softmax"
24 name: "Softmax"
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | softmax_quantized_test.cc | 55 optimized_ops::Softmax(sm_params, shape_common, reference_dequant_data.data(), in RunSoftmaxFloatReference()
135 optimized_ops::Softmax(params, shape_common, input_data, shape_common, in RunOneSoftmaxTest()
137 reference_ops::Softmax(params, shape_common, input_data, shape_common, in RunOneSoftmaxTest()
205 optimized_integer_ops::Softmax(params, shape_common, input_data, shape_common, in RunOneSoftmaxTestInt8()
207 reference_integer_ops::Softmax(params, shape_common, input_data, shape_common, in RunOneSoftmaxTestInt8()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | softmax.cc | 36 class Softmax : public NodeShader { class
96 return absl::make_unique<Softmax>(); in NewSoftmaxNodeShader()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 28 inline void Softmax(const SoftmaxParams& params, in Softmax() function
64 inline void Softmax(const SoftmaxParams& params, in Softmax() function
139 inline void Softmax(const float* in, const int input_size, const int batch_size, in Softmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | softmax_test.cc | 98 tflite::reference_ops::Softmax(params, input_shape, input_buffer, input_shape, in TEST()
127 tflite::reference_ops::Softmax(params, input_shape, input_buffer, input_shape, in TEST()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa-hifimini/ |
| D | softmax.cc | 36 inline void Softmax(const SoftmaxParams& params, in Softmax() function
176 xtensa::hifimini::Softmax(op_params, shape, GetTensorData<int8_t>(input), in Softmax2DQuantized()
180 xtensa::hifimini::Softmax(op_params, shape, GetTensorData<int8_t>(input), in Softmax2DQuantized()
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_Softmax.pbtxt | 2 graph_op_name: "Softmax"
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | advanced_activations.py | 263 class Softmax(Layer): class
279 super(Softmax, self).__init__(**kwargs)
288 base_config = super(Softmax, self).get_config()
|
| D | advanced_activations_test.py | 60 testing_utils.layer_test(keras.layers.Softmax,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | softmax.h | 30 std::vector<ComputeTaskDescriptorPtr> Softmax(int id, ValueId input_id,
|
| D | softmax.cc | 99 std::vector<ComputeTaskDescriptorPtr> Softmax(int id, ValueId input_id, in Softmax() function
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Softmax.pbtxt | 2 graph_op_name: "Softmax"
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | keras_imagenet_main_graph_mode.golden_summary | 44 Softmax 1
|
| D | keras_imagenet_main.golden_summary | 43 Softmax 1
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/selectors/ |
| D | simple_selectors.cc | 178 Softmax operation = CreateSoftmax(op_def); in SelectSoftmax()
179 *ptr = absl::make_unique<Softmax>(std::move(operation)); in SelectSoftmax()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.-softmax.pbtxt | 1 path: "tensorflow.keras.layers.Softmax"
3 is_instance: "<class \'tensorflow.python.keras.layers.advanced_activations.Softmax\'>"
|
| /external/XNNPACK/ |
| D | README.md | 36 - Softmax
84 - Marat Dukhan, Artsiom Ablavatski "The Two-Pass Softmax Algorithm".
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-softmax.pbtxt | 1 path: "tensorflow.keras.layers.Softmax"
3 is_instance: "<class \'tensorflow.python.keras.layers.advanced_activations.Softmax\'>"
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | softmax.h | 25 inline void Softmax(const SoftmaxParams& params, in Softmax() function
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator_test.cc | 81 auto softmax = ops::Softmax(s.WithOpName("softmax"), logits); in CreateMiniGraph()
|