Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize_parameterized_test.py164 def _TestQuantize_Conv2dWithoutBatchNorm(self, activation, activation_op_name, argument
218 def _TestQuantize_FCWithoutBatchNorm(self, activation, activation_op_name, argument
265 self, activation, activation_op_name, with_bypass, delay, use_resource, argument
315 def _TestQuantize_AtrousConvWithoutBatchNorm(self, activation, argument
489 def _TestQuantize_Conv2dWithBatchNorm(self, activation, activation_op_name, argument
546 def _TestQuantize_FCWithBatchNorm(self, activation, activation_op_name, argument
603 self, activation, activation_op_name, with_bypass, delay, argument
661 self, activation, activation_op_name, with_bypass, delay, argument
739 def _TestBatchNormForcedUpdates(self, activation, activation_op_name, argument
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc516 ScopedActivateContext activation(context); in ContextGetSharedMemConfig() local
533 ScopedActivateContext activation(context); in ContextSetSharedMemConfig() local
555 ScopedActivateContext activation(context); in LaunchKernel() local
575 ScopedActivateContext activation(context); in LoadCubin() local
592 ScopedActivateContext activation(context); in LoadPtx() local
665 ScopedActivateContext activation(context); in SynchronousMemsetUint8() local
678 ScopedActivateContext activation(context); in SynchronousMemsetUint32() local
692 ScopedActivateContext activation(context); in AsynchronousMemsetUint8() local
707 ScopedActivateContext activation(context); in AsynchronousMemsetUint32() local
848 ScopedActivateContext activation(context); in DeviceDeallocate() local
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc404 ScopedActivateContext activation{context}; in ContextGetSharedMemConfig() local
420 ScopedActivateContext activation{context}; in ContextSetSharedMemConfig() local
441 ScopedActivateContext activation{context}; in LaunchKernel() local
479 ScopedActivateContext activation{context}; in LoadHsaco() local
502 ScopedActivateContext activation{context}; in SynchronousMemsetUint8() local
515 ScopedActivateContext activation{context}; in SynchronousMemsetUint32() local
530 ScopedActivateContext activation{context}; in AsynchronousMemsetUint8() local
546 ScopedActivateContext activation{context}; in AsynchronousMemsetUint32() local
672 ScopedActivateContext activation{context}; in DeviceDeallocate() local
700 ScopedActivateContext activation{context}; in HostAllocate() local
[all …]
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h76 TfLiteFusedActivation activation; member
85 TfLiteFusedActivation activation; member
97 TfLiteFusedActivation activation; member
105 TfLiteFusedActivation activation; member
109 TfLiteFusedActivation activation; member
114 TfLiteFusedActivation activation; member
119 TfLiteFusedActivation activation; member
130 TfLiteFusedActivation activation; member
152 TfLiteFusedActivation activation; member
156 TfLiteFusedActivation activation; member
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc42 const TfLiteFusedActivation& activation, int32_t* multiplier, int* shift, in PopulateConvolutionQuantizationParams()
128 void CalculateActivationRangeQuantizedImpl(TfLiteFusedActivation activation, in CalculateActivationRangeQuantizedImpl()
156 TfLiteFusedActivation activation, in CalculateActivationRangeQuantized()
180 void CalculateActivationRangeUint8(TfLiteFusedActivation activation, in CalculateActivationRangeUint8()
190 void CalculateActivationRangeInt8(TfLiteFusedActivation activation, in CalculateActivationRangeInt8()
Dkernel_util.h125 void CalculateActivationRange(TfLiteFusedActivation activation, in CalculateActivationRange()
/external/tensorflow/tensorflow/python/layers/
Dconvolutional.py86 activation=None, argument
129 activation=None, argument
285 activation=None, argument
328 activation=None, argument
492 activation=None, argument
535 activation=None, argument
702 activation=None, argument
814 activation=None, argument
866 activation=None, argument
986 activation=None, argument
[all …]
Dcore.py86 activation=None, argument
119 activation=None, argument
/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc26 TfLiteFusedActivation activation, in RnnBatchStep()
42 TfLiteFusedActivation activation, in RnnBatchStep()
125 TfLiteFusedActivation activation, int8_t* quantized_input_ptr_batch, in RnnBatchStep()
147 TfLiteFusedActivation activation, int8_t* quantized_input_ptr_batch, in RnnBatchStep()
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Drnn_cells.py59 activation=None, argument
191 activation=None, argument
Dcore_layers.py94 activation=None, argument
300 activation=None, argument
375 activation=None, argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Delu_op.cc52 const auto activation = ctx->Input(1); in Compile() local
93 const auto activation = ctx->Input(1); in Compile() local
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dmodel_utils.py37 activation=nn_ops.relu, argument
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_v2.py151 activation='tanh', argument
321 def standard_gru(inputs, init_h, kernel, recurrent_kernel, bias, activation, argument
520 activation='tanh', argument
715 activation, recurrent_activation, time_major): argument
Dconvolutional.py106 activation=None, argument
341 activation=None, argument
457 activation=None, argument
575 activation=None, argument
703 activation=None, argument
977 activation=None, argument
1237 activation=None, argument
1440 activation=None, argument
1609 activation=None, argument
1742 activation=None, argument
Dlocal.py125 activation=None, argument
400 activation=None, argument
Drecurrent.py1171 activation='tanh', argument
1357 activation='tanh', argument
1419 def activation(self): member in SimpleRNN
1571 activation='tanh', argument
1872 activation='tanh', argument
1939 def activation(self): member in GRU
2112 activation='tanh', argument
2484 activation='tanh', argument
2551 def activation(self): member in LSTM
Dconvolutional_recurrent.py556 activation='tanh', argument
884 activation='tanh', argument
965 def activation(self): member in ConvLSTM2D
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py49 activation=None, argument
201 activation=None, argument
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Dcore_rnn_cell.py282 activation=None, argument
340 def __init__(self, cell, output_size, activation=None, reuse=None): argument
Drnn_cell.py145 activation=math_ops.tanh, argument
1366 activation=math_ops.tanh, argument
1642 activation=math_ops.tanh, argument
2270 activation=math_ops.tanh, argument
2506 activation=None, argument
2763 def __init__(self, num_units, activation=None, reuse=None, name=None, argument
2852 activation=None, argument
3100 activation=None, argument
3190 activation=None, argument
3304 activation=None, argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc203 auto activation = convolution->mutable_operand(0); in HandleBatchGroupCount() local
403 auto activation = convolution->mutable_operand(0); in HandleConvolution() local
475 auto activation = convolution->mutable_operand(0); in HandleConvolution() local
/external/tensorflow/tensorflow/python/keras/
Dactivations.py238 def serialize(activation): argument
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dreal_nvp.py256 activation=nn_ops.relu, argument
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv_test.cc35 TfLiteFusedActivation activation, in TestDepthwiseConvFloat()
117 float output_max, TfLiteFusedActivation activation, uint8_t* output_data) { in TestDepthwiseConvQuantized()

12