Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc504 ScopedActivateContext activation(context); in ContextGetSharedMemConfig() local
512 ScopedActivateContext activation(context); in ContextSetSharedMemConfig() local
524 ScopedActivateContext activation(context); in LaunchKernel() local
540 ScopedActivateContext activation(context); in LoadCubin() local
553 ScopedActivateContext activation(context); in LoadPtx() local
626 ScopedActivateContext activation(context); in SynchronousMemsetUint8() local
635 ScopedActivateContext activation(context); in SynchronousMemsetUint32() local
644 ScopedActivateContext activation(context); in AsynchronousMemsetUint8() local
654 ScopedActivateContext activation(context); in AsynchronousMemsetUint32() local
802 ScopedActivateContext activation(context); in DeviceDeallocate() local
[all …]
Dcuda_asm_compiler.cc37 gpu::ScopedActivateContext activation(context); in LinkGpuAsm() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc382 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
620 ScopedActivateContext activation{context}; in DeviceDeallocate() local
648 ScopedActivateContext activation{context}; in HostAllocate() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dactivation_layer_builder.h31 TfLiteFusedActivation activation) in ActivationLayerBuilder()
38 void SetActivation(TfLiteFusedActivation activation) { in SetActivation()
Dfully_connected_op_builder.cc90 TfLiteFusedActivation activation = fc_params->activation; in PopulateSubgraph() local
180 TfLiteFusedActivation activation = fc_params->activation; in IsFullyConnectedOpSupported() local
Dconvolution_op_builder.cc208 TfLiteFusedActivation activation; in PopulateSubgraph() local
303 TfLiteFusedActivation activation; in IsConvolutionOpSupported() local
Dadd_op_builder.cc48 TfLiteFusedActivation activation = params->activation; in PopulateSubgraph() local
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py90 activation=None, argument
130 activation=None, argument
295 activation=None, argument
335 activation=None, argument
508 activation=None, argument
548 activation=None, argument
724 activation=None, argument
839 activation=None, argument
888 activation=None, argument
1011 activation=None, argument
[all …]
Dcore.py86 activation=None, argument
116 activation=None, argument
/external/tensorflow/tensorflow/lite/micro/kernels/
Dfully_connected_common.cc48 TfLiteFusedActivation activation) { in FullyConnectedParamsFloat()
56 TfLiteContext* context, TfLiteFusedActivation activation, in CalculateOpDataFullyConnected()
Dmul_test.cc46 TfLiteFusedActivation activation, const T* golden, in ValidateMulGoldens()
73 float* output_data, TfLiteFusedActivation activation) { in TestMulFloat()
100 TfLiteFusedActivation activation) { in TestMulQuantized()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h82 TfLiteFusedActivation activation; member
98 TfLiteFusedActivation activation; member
107 TfLiteFusedActivation activation; member
129 TfLiteFusedActivation activation; member
137 TfLiteFusedActivation activation; member
144 TfLiteFusedActivation activation; member
152 TfLiteFusedActivation activation; member
160 TfLiteFusedActivation activation; member
174 TfLiteFusedActivation activation; member
207 TfLiteFusedActivation activation; member
[all …]
/external/tensorflow/tensorflow/python/keras/applications/
Dinception_resnet_v2.py258 activation='relu', argument
295 def inception_resnet_block(x, scale, block_type, block_idx, activation='relu'): argument
Dmobilenet_v3.py365 def stack_fn(x, kernel, activation, se_ratio): argument
401 def stack_fn(x, kernel, activation, se_ratio): argument
490 activation, block_id): argument
/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/tensorflow/tensorflow/python/keras/layers/ops/
Dcore.py30 def dense(inputs, kernel, bias=None, activation=None, dtype=None): argument
/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/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc34 TfLiteTensor* output_state, TfLiteTensor* activation, in GruImpl()
199 TfLiteTensor* activation; in Prepare() local
248 TfLiteTensor* activation; in Eval() local
/external/tensorflow/tensorflow/core/protobuf/
Dconv_autotuning.proto26 stream_executor.dnn.ActivationMode activation = 8; field
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py122 activation=None, argument
492 activation=None, argument
651 activation=None, argument
798 activation=None, argument
926 activation=None, argument
1197 activation=None, argument
1507 activation=None, argument
1772 activation=None, argument
2005 activation=None, argument
2195 activation=None, argument
[all …]
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py51 activation=None, argument
208 activation=None, argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc106 int activation = kMissingIndex; member
130 int activation = kMissingIndex; member
170 int activation = kMissingIndex; member
206 int activation = kMissingIndex; member
966 const NodeDef* activation = nullptr) { in IsCpuCompatibleDataType()
988 const NodeDef* activation = nullptr) { in IsCpuCompatibleDataType()
1034 const NodeDef* activation = nullptr) { in IsCpuCompatibleDataType()
1112 const NodeDef& activation = graph->node(matched.activation); in IsCpuCompatibleDataType() local
1250 const NodeDef& activation = graph->node(matched.activation); in IsCpuCompatibleDataType() local
1341 const NodeDef& activation = graph->node(matched.activation); in IsCpuCompatibleDataType() local
[all …]
/external/tensorflow/tensorflow/python/keras/premade/
Dlinear.py67 activation=None, argument

1234