| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_LogSoftmax.pbtxt | 2 graph_op_name: "LogSoftmax"
4 name: "nn.LogSoftmax"
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | logsoftmax_quantized_test.cc | 54 optimized_ops::LogSoftmax(sm_params, shape_common, in RunLogSoftmaxFloatReference()
88 optimized_ops::LogSoftmax(sm_params, shape_common, in RunLogSoftmaxFloatReference()
179 optimized_ops::LogSoftmax(params, shape_common, input_data, shape_common, in RunOneLogSoftmaxTest()
181 reference_ops::LogSoftmax(params, shape_common, input_data, shape_common, in RunOneLogSoftmaxTest()
227 reference_integer_ops::LogSoftmax( in RunOneLogSoftmaxTest()
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_LogSoftmax.pbtxt | 2 graph_op_name: "LogSoftmax"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_LogSoftmax.pbtxt | 2 graph_op_name: "LogSoftmax"
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | log_softmax_test.cc | 95 tflite::reference_ops::LogSoftmax(params, input_shape, input_buffer, in TEST()
|
| D | activations.cc | 835 optimized_ops::LogSoftmax( in LogSoftmaxEval()
839 reference_ops::LogSoftmax( in LogSoftmaxEval()
853 optimized_ops::LogSoftmax( in LogSoftmaxEval()
857 reference_ops::LogSoftmax( in LogSoftmaxEval()
871 reference_integer_ops::LogSoftmax( in LogSoftmaxEval()
|
| D | activations_test.cc | 758 TEST(FloatActivationsOpTest, LogSoftmax) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | log_softmax.h | 23 inline void LogSoftmax(int32_t input_multiplier, int32_t input_shift, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad_test.cc | 37 using ops::LogSoftmax;
132 auto y = LogSoftmax(scope_, x); in TEST_F()
|
| D | nn_grad.cc | 108 auto minus_log_softmax = Multiply(scope, LogSoftmax(scope, logits), -1.0f); in SoftmaxCrossEntropyWithLogitsGrad()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1461 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
1465 LogSoftmax(params, input_shape, input_data, output_shape, output_data); in LogSoftmax()
1468 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
1470 LogSoftmax(input_data, DimsToShape(input_dims), output_data, in LogSoftmax()
1474 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
1485 LogSoftmax(params, input_shape, input_data, output_shape, output_data); in LogSoftmax()
1488 inline void LogSoftmax(const uint8* input_data, const Dims<4>& input_dims, in LogSoftmax() function
1493 LogSoftmax(input_data, DimsToShape(input_dims), input_multiplier, in LogSoftmax()
|
| D | optimized_ops.h | 4664 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
4700 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 717 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
721 LogSoftmax(params, input_shape, input_data, output_shape, output_data); in LogSoftmax()
724 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
735 LogSoftmax(params, input_shape, input_data, output_shape, output_data); in LogSoftmax()
1693 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
1695 LogSoftmax(input_data, DimsToShape(input_dims), output_data, in LogSoftmax()
1699 inline void LogSoftmax(const uint8* input_data, const Dims<4>& input_dims, in LogSoftmax() function
1704 LogSoftmax(input_data, DimsToShape(input_dims), input_multiplier, in LogSoftmax()
|
| D | reference_ops.h | 2567 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
2599 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1830 class LogSoftmax : public SimpleOperator<LogSoftmaxOperator> { class
1832 explicit LogSoftmax() in LogSoftmax() function in toco::tflite::LogSoftmax
2513 ops.push_back(MakeUnique<LogSoftmax>()); in BuildOperatorList()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 123 `LogSoftmax` | `T={double,float}`
|
| D | cpu_supported_ops.md | 123 `LogSoftmax` | `T={double,float}`
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4802 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
4839 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
4843 LogSoftmax(params, input_shape, input_data, output_shape, output_data); in LogSoftmax()
4960 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
5052 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
5063 LogSoftmax(params, input_shape, input_data, output_shape, output_data); in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 341 HANDLE_OPERATORTYPENAME_CASE(LogSoftmax) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2191 TEST_F(OpTest, LogSoftmax) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1776 name: "LogSoftmax"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1776 name: "LogSoftmax"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 15603 name: "LogSoftmax"
15624 name: "LogSoftmax"
|
| D | ops_history.v1.pbtxt | 33835 name: "LogSoftmax"
33857 name: "LogSoftmax"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 8257 func LogSoftmax(scope *Scope, logits tf.Output) (logsoftmax tf.Output) { func
|