Home
last modified time | relevance | path

Searched defs:activation (Results 1 – 25 of 96) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_driver.cc445 ScopedActivateContext activation(context); in ContextGetSharedMemConfig() local
453 ScopedActivateContext activation(context); in ContextSetSharedMemConfig() local
465 ScopedActivateContext activation(context); in LaunchKernel() local
484 ScopedActivateContext activation(context); in LoadCubin() local
497 ScopedActivateContext activation(context); in LoadPtx() local
570 ScopedActivateContext activation(context); in SynchronousMemsetUint8() local
579 ScopedActivateContext activation(context); in SynchronousMemsetUint32() local
588 ScopedActivateContext activation(context); in AsynchronousMemsetUint8() local
598 ScopedActivateContext activation(context); in AsynchronousMemsetUint32() local
746 ScopedActivateContext activation(context); in DeviceDeallocate() local
[all …]
Dcuda_asm_compiler.cc48 gpu::ScopedActivateContext activation(context); in LinkGpuAsm() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc388 ScopedActivateContext activation{context}; in ContextGetSharedMemConfig() local
397 ScopedActivateContext activation{context}; in ContextSetSharedMemConfig() local
410 ScopedActivateContext activation{context}; in LaunchKernel() local
447 ScopedActivateContext activation{context}; in LoadHsaco() local
468 ScopedActivateContext activation{context}; in SynchronousMemsetUint8() local
477 ScopedActivateContext activation{context}; in SynchronousMemsetUint32() local
488 ScopedActivateContext activation{context}; in AsynchronousMemsetUint8() local
498 ScopedActivateContext activation{context}; in AsynchronousMemsetUint32() local
628 ScopedActivateContext activation{context}; in DeviceDeallocate() local
657 ScopedActivateContext activation{context}; in HostAllocate() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dactivation_layer_builder.h33 TfLiteFusedActivation activation) in ActivationLayerBuilder()
40 void SetActivation(TfLiteFusedActivation activation) { in SetActivation()
Dfully_connected_op_builder.cc94 TfLiteFusedActivation activation = fc_params->activation; in PopulateSubgraph() local
184 TfLiteFusedActivation activation = fc_params->activation; in IsFullyConnectedOpSupported() local
Dconvolution_op_builder.cc212 TfLiteFusedActivation activation; in PopulateSubgraph() local
307 TfLiteFusedActivation activation; in IsConvolutionOpSupported() local
Dmul_op_builder.cc57 TfLiteFusedActivation activation = params->activation; in PopulateSubgraph() local
Dadd_op_builder.cc51 TfLiteFusedActivation activation = params->activation; in PopulateSubgraph() local
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py87 activation=None, argument
128 activation=None, argument
294 activation=None, argument
335 activation=None, argument
509 activation=None, argument
550 activation=None, argument
727 activation=None, argument
843 activation=None, argument
893 activation=None, argument
1017 activation=None, argument
[all …]
Dcore.py84 activation=None, argument
115 activation=None, argument
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h81 TfLiteFusedActivation activation; member
97 TfLiteFusedActivation activation; member
108 TfLiteFusedActivation activation; member
130 TfLiteFusedActivation activation; member
138 TfLiteFusedActivation activation; member
145 TfLiteFusedActivation activation; member
153 TfLiteFusedActivation activation; member
161 TfLiteFusedActivation activation; member
175 TfLiteFusedActivation activation; member
208 TfLiteFusedActivation activation; member
[all …]
/external/ComputeLibrary/tests/validation/NEON/
DActivationLayer.cpp62 …ance<float> relative_tolerance(DataType data_type, ActivationLayerInfo::ActivationFunction activat… in relative_tolerance()
108 …ance<float> absolute_tolerance(DataType data_type, ActivationLayerInfo::ActivationFunction activat… in absolute_tolerance()
151 AbsoluteTolerance<uint8_t> tolerance_qasymm8(ActivationLayerInfo::ActivationFunction activation) in tolerance_qasymm8()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc28 TfLiteFusedActivation activation, in RnnBatchStep()
44 TfLiteFusedActivation activation, in RnnBatchStep()
124 TfLiteFusedActivation activation, int8_t* quantized_input_ptr_batch, in RnnBatchStep()
150 TfLiteFusedActivation activation, int8_t* quantized_input_ptr_batch, in RnnBatchStep()
/external/rnnoise/src/
Drnn.h49 int activation; member
58 int activation; member
/external/rust/crates/grpcio-sys/grpc/src/core/lib/security/authorization/mock_cel/
Devaluator_core.h59 const BaseActivation& activation) const override { in Evaluate()
/external/ComputeLibrary/tests/validation/reference/
DActivationLayer.h39 inline T activate_float(T x, T a, T b, ActivationLayerInfo::ActivationFunction activation) in activate_float()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Delu_op.cc67 const auto activation = ctx->Input(1); in Compile() local
99 const auto activation = ctx->Input(1); in Compile() local
/external/tensorflow/tensorflow/core/transforms/remapper/
Dremapping_helper.h43 Operation* activation; member
56 Operation* activation; member
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py117 activation=None, argument
494 activation=None, argument
654 activation=None, argument
801 activation=None, argument
929 activation=None, argument
1200 activation=None, argument
1510 activation=None, argument
1775 activation=None, argument
2008 activation=None, argument
2199 activation=None, argument
[all …]
Drecurrent.py1303 activation='tanh', argument
1527 activation='tanh', argument
1595 def activation(self): member in SimpleRNN
1742 activation='tanh', argument
2064 activation='tanh', argument
2137 def activation(self): member in GRU
2305 activation='tanh', argument
2563 activation='tanh', argument
2737 activation='tanh', argument
2810 def activation(self): member in LSTM
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_gru.cc36 TfLiteTensor* output_state, TfLiteTensor* activation, in GruImpl()
207 TfLiteTensor* activation; in Prepare() local
256 TfLiteTensor* activation; in Eval() local
Dkernel_util.cc200 const TfLiteFusedActivation& activation, int32_t* multiplier, int* shift, in PopulateConvolutionQuantizationParams()
215 const TfLiteFusedActivation& activation, int32_t* multiplier, int* shift, in PopulateConvolutionQuantizationParams()
352 TfLiteContext* context, TfLiteFusedActivation activation, int32_t qmin, in CalculateActivationRangeQuantizedImpl()
386 TfLiteFusedActivation activation, in CalculateActivationRangeQuantized()
Dconv3d_test.cc36 ActivationFunctionType activation = ActivationFunctionType_NONE, in Conv3dOpModel()
56 ActivationFunctionType activation = ActivationFunctionType_NONE, in Conv3dOpModel()
/external/tensorflow/tensorflow/core/protobuf/
Dconv_autotuning.proto26 stream_executor.dnn.ActivationMode activation = 8; field
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfused_kernel_matcher.cc158 auto activation = GetActivation(bias_add); in matchAndRewrite() local
270 auto activation = GetActivation(bias_add); in IsDeviceCompatible() local

1234