| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 382 ScopedActivateContext activation{context}; in ContextGetSharedMemConfig() local
391 ScopedActivateContext activation{context}; in ContextSetSharedMemConfig() local
404 ScopedActivateContext activation{context}; in LaunchKernel() local
439 ScopedActivateContext activation{context}; in LoadHsaco() local
460 ScopedActivateContext activation{context}; in SynchronousMemsetUint8() local
469 ScopedActivateContext activation{context}; in SynchronousMemsetUint32() local
480 ScopedActivateContext activation{context}; in AsynchronousMemsetUint8() local
490 ScopedActivateContext activation{context}; in AsynchronousMemsetUint32() local
613 ScopedActivateContext activation{context}; in DeviceDeallocate() local
641 ScopedActivateContext activation{context}; in HostAllocate() local
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 486 ScopedActivateContext activation(context); in ContextGetSharedMemConfig() local
494 ScopedActivateContext activation(context); in ContextSetSharedMemConfig() local
506 ScopedActivateContext activation(context); in LaunchKernel() local
522 ScopedActivateContext activation(context); in LoadCubin() local
535 ScopedActivateContext activation(context); in LoadPtx() local
608 ScopedActivateContext activation(context); in SynchronousMemsetUint8() local
617 ScopedActivateContext activation(context); in SynchronousMemsetUint32() local
626 ScopedActivateContext activation(context); in AsynchronousMemsetUint8() local
636 ScopedActivateContext activation(context); in AsynchronousMemsetUint32() local
776 ScopedActivateContext activation(context); in DeviceDeallocate() local
[all …]
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 82 TfLiteFusedActivation activation; member
96 TfLiteFusedActivation activation; member
118 TfLiteFusedActivation activation; member
126 TfLiteFusedActivation activation; member
130 TfLiteFusedActivation activation; member
135 TfLiteFusedActivation activation; member
140 TfLiteFusedActivation activation; member
151 TfLiteFusedActivation activation; member
179 TfLiteFusedActivation activation; member
183 TfLiteFusedActivation activation; member
[all …]
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | convolutional.py | 86 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 …]
|
| D | core.py | 86 activation=None, argument
118 activation=None, argument
|
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | inception_resnet_v2.py | 238 activation='relu', argument
275 def inception_resnet_block(x, scale, block_type, block_idx, activation='relu'): argument
|
| D | efficientnet.py | 150 activation='swish', argument
352 activation='swish', argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 147 activation='tanh', argument
318 activation='tanh', argument
505 def standard_gru(inputs, init_h, kernel, recurrent_kernel, bias, activation, argument
674 mask, time_major, go_backwards, activation, argument
726 mask, time_major, go_backwards, activation, argument
875 activation='tanh', argument
1030 activation='tanh', argument
1239 activation, recurrent_activation, mask, time_major, argument
1433 go_backwards, activation, recurrent_activation, argument
1486 bias, mask, time_major, go_backwards, activation, argument
|
| D | convolutional.py | 109 activation=None, argument
417 activation=None, argument
570 activation=None, argument
714 activation=None, argument
862 activation=None, argument
1155 activation=None, argument
1413 activation=None, argument
1641 activation=None, argument
1826 activation=None, argument
1973 activation=None, argument
|
| D | local.py | 140 activation=None, argument
454 activation=None, argument
|
| D | recurrent.py | 1267 activation='tanh', argument
1485 activation='tanh', argument
1554 def activation(self): member in SimpleRNN
1707 activation='tanh', argument
2025 activation='tanh', argument
2099 def activation(self): member in GRU
2273 activation='tanh', argument
2671 activation='tanh', argument
2745 def activation(self): member in LSTM
|
| D | convolutional_recurrent.py | 557 activation='tanh', argument
885 activation='tanh', argument
966 def activation(self): member in ConvLSTM2D
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 28 TfLiteFusedActivation activation, in RnnBatchStep()
44 TfLiteFusedActivation activation, in RnnBatchStep()
126 TfLiteFusedActivation activation, int8_t* quantized_input_ptr_batch, in RnnBatchStep()
148 TfLiteFusedActivation activation, int8_t* quantized_input_ptr_batch, in RnnBatchStep()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 30 const TfLiteFusedActivation& activation, int32_t* multiplier, int* shift, in PopulateConvolutionQuantizationParams()
45 const TfLiteFusedActivation& activation, int32_t* multiplier, int* shift, in PopulateConvolutionQuantizationParams()
149 void CalculateActivationRangeQuantizedImpl(TfLiteFusedActivation activation, in CalculateActivationRangeQuantizedImpl()
177 TfLiteFusedActivation activation, in CalculateActivationRangeQuantized()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | elu_op.cc | 67 const auto activation = ctx->Input(1); in Compile() local
99 const auto activation = ctx->Input(1); in Compile() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | conv_autotuning.proto | 26 stream_executor.dnn.ActivationMode activation = 8; field
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru.cc | 34 TfLiteTensor* output_state, TfLiteTensor* activation, in GruImpl()
186 TfLiteTensor* activation = GetTemporary(context, node, kActivation); in Prepare() local
219 TfLiteTensor* activation = GetTemporary(context, node, kActivation); in Eval() local
|
| D | gru_cell.cc | 41 const RuntimeShape& activation_shape, float* activation, in GruCell()
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 50 activation=None, argument
203 activation=None, argument
|
| /external/tensorflow/tensorflow/python/keras/premade/ |
| D | linear.py | 65 activation=None, argument
|
| D | wide_deep.py | 72 def __init__(self, linear_model, dnn_model, activation=None, **kwargs): argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm.cc | 37 TensorCodeGenerator activation("dst_data", state_size, op_def.dst_tensors[0]); in GetLSTMCode() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 95 int activation = kMissingIndex; member
119 int activation = kMissingIndex; member
159 int activation = kMissingIndex; member
195 int activation = kMissingIndex; member
954 const NodeDef& activation = graph->node(matched.activation); in AddFusedContractionNode() local
1087 const NodeDef& activation = graph->node(matched.activation); in AddFusedConv2DNode() local
1170 const NodeDef& activation = graph->node(matched.activation); in AddFusedContractionNode() local
1210 const NodeDef& activation = graph->node(matched.activation); in AddFusedBatchNormExNode() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | activations.py | 359 def serialize(activation): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_cell_impl.py | 415 activation=None, argument
519 activation=None, argument
669 activation=None, argument
852 activation=None, argument
|