| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op_functor.h | 34 typename TTypes<T>::Tensor activations) { in operator()
35 activations.device(d) = in operator()
68 typename TTypes<T>::Tensor activations) { in operator()
69 activations.device(d) = in operator()
111 typename TTypes<T>::Tensor activations; member
115 args.activations.device(args.d) = in operator()
146 typename TTypes<T>::Tensor activations) { in operator()
148 activations.device(d) = in operator()
164 typename TTypes<T>::ConstTensor activations, in operator()
167 (activations < static_cast<T>(0)) in operator()
[all …]
|
| D | relu_op.cc | 82 typename TTypes<T>::Tensor activations); \
95 typename TTypes<T>::Tensor activations); \
119 typename TTypes<T>::Tensor activations); \
125 typename TTypes<T>::ConstTensor activations, \
132 typename TTypes<T>::Tensor activations); \
138 typename TTypes<T>::ConstTensor activations, \
146 typename TTypes<qint8>::Tensor activations);
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | activations_test.py | 24 from tensorflow.python.keras import activations
49 fn = activations.get(name)
50 ref_fn = getattr(activations, name)
52 config = activations.serialize(fn)
53 fn = activations.deserialize(config)
59 fn_v2 = activations.get(fn_v2_key)
60 config = activations.serialize(fn_v2)
61 fn = activations.deserialize(config)
84 f = backend.function([x], [activations.softmax(x)])
93 activations.softmax(x)
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 52 namespace activations { namespace
1451 static TfLiteRegistration r = {activations::Init, activations::Free, in Register_ELU()
1452 activations::EluPrepare, activations::EluEval}; in Register_ELU()
1457 static TfLiteRegistration r = {activations::ReluInit, activations::ReluFree, in Register_RELU()
1458 activations::ReluPrepare, in Register_RELU()
1459 activations::ReluEval}; in Register_RELU()
1464 static TfLiteRegistration r = {activations::ReluInit, activations::ReluFree, in Register_RELU_N1_TO_1()
1465 activations::ReluPrepare, in Register_RELU_N1_TO_1()
1466 activations::Relu1Eval}; in Register_RELU_N1_TO_1()
1471 static TfLiteRegistration r = {activations::ReluInit, activations::ReluFree, in Register_RELU6()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 117 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers,
125 HloInstruction* activations, int64 spatial_dimension_to_split,
162 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers,
169 HloInstruction* activations, int64 batch_dimension, int64 old_batch_size,
177 HloInstruction* activations, int64 spatial_dimension_to_split,
344 HloInstruction* activations, int64 spatial_dimension_to_split, in HaloDuplicateWithSlice() argument
349 activations->shape().dimensions(activations_batch_dim) / kNumSplits; in HaloDuplicateWithSlice()
352 std::vector<int64> new_dimensions(activations->shape().dimensions().begin(), in HaloDuplicateWithSlice()
353 activations->shape().dimensions().end()); in HaloDuplicateWithSlice()
359 TF_ASSIGN_OR_RETURN(activations, in HaloDuplicateWithSlice()
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_gradient.py | 32 def get_gradients_through_compute_gradients(optimizer, loss, activations): argument
45 activation_list = activations.values()
49 zip(activations.keys(), grads))
103 def hook_dummy_table_variables_to_activations(tpu_embedding, activations, argument
119 for feature in activations:
123 activations[feature],
|
| D | tpu_embedding_v2_correctness_test.py | 176 activations = mid_level_api.dequeue()
177 total_loss = _get_total_loss_tensor(activations)
178 ret_val = [total_loss] + list(activations)
182 activations = mid_level_api.dequeue()
183 tape.watch(activations)
184 total_loss = _get_total_loss_tensor(activations)
186 gradients = tape.gradient(loss_per_replica, activations)
188 ret_val = [total_loss] + list(activations)
508 activations = mid_level.dequeue()
510 activations))
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Softplus.pbtxt | 8 name: "activations"
36 name: "activations"
66 name: "activations"
97 name: "activations"
128 name: "activations"
|
| D | Softsign.pbtxt | 8 name: "activations"
36 name: "activations"
66 name: "activations"
97 name: "activations"
128 name: "activations"
|
| D | Relu.pbtxt | 8 name: "activations"
36 name: "activations"
66 name: "activations"
97 name: "activations"
128 name: "activations"
|
| D | Relu6.pbtxt | 8 name: "activations"
36 name: "activations"
66 name: "activations"
97 name: "activations"
|
| D | Elu.pbtxt | 8 name: "activations"
30 name: "activations"
52 name: "activations"
|
| D | Selu.pbtxt | 8 name: "activations"
30 name: "activations"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Softplus.pbtxt | 8 name: "activations"
36 name: "activations"
66 name: "activations"
97 name: "activations"
128 name: "activations"
|
| D | Softsign.pbtxt | 8 name: "activations"
36 name: "activations"
66 name: "activations"
97 name: "activations"
128 name: "activations"
|
| D | Relu.pbtxt | 8 name: "activations"
36 name: "activations"
66 name: "activations"
97 name: "activations"
128 name: "activations"
|
| D | Relu6.pbtxt | 8 name: "activations"
36 name: "activations"
66 name: "activations"
97 name: "activations"
|
| D | Elu.pbtxt | 8 name: "activations"
30 name: "activations"
52 name: "activations"
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | README.md | 156 * Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
157 but fused `TANH` and `SIGN_BIT` activations are not.
163 * Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
164 but fused `TANH` and `SIGN_BIT` activations are not.
175 * Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
176 but fused `TANH` and `SIGN_BIT` activations are not.
188 * Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
189 but fused `TANH` and `SIGN_BIT` activations are not.
194 * Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
195 but fused `TANH` and `SIGN_BIT` activations are not.
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 100 HloInstruction* activations = in TEST_F() local
111 activations->shape(), gradients->shape(), /*feature_group_count=*/1, in TEST_F()
116 activations, gradients, /*feature_group_count=*/1, in TEST_F()
142 HloInstruction* activations = in TEST_F() local
152 activations->shape(), gradients->shape(), /*feature_group_count=*/1, in TEST_F()
157 activations, gradients, /*feature_group_count=*/1, in TEST_F()
173 HloInstruction* activations = in TEST_F() local
187 ShapeUtil::MakeShape(F32, {32, 3, 3, 32}), activations, gradients, in TEST_F()
203 HloInstruction* activations = in TEST_F() local
217 ShapeUtil::MakeShape(F32, {320, 3, 3, 192}), activations, gradients, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RecvTPUEmbeddingActivations.pbtxt | 7 A TensorList of embedding activations containing one Tensor per
24 summary: "An op that receives embedding activations on the TPU."
30 one Tensor of activations per table specified in the model. There can be at
|
| D | api_def_TPUEmbeddingActivations.pbtxt | 13 The embedding activations Tensor to return.
20 these activations were computed.
27 activations.
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | activations.cc | 30 namespace activations { namespace
265 return {/*init=*/activations::ReluInit, in Register_RELU()
267 /*prepare=*/activations::ReluPrepare, in Register_RELU()
268 /*invoke=*/activations::ReluEval, in Register_RELU()
276 return {/*init=*/activations::Relu6Init, in Register_RELU6()
278 /*prepare=*/activations::Relu6Prepare, in Register_RELU6()
279 /*invoke=*/activations::Relu6Eval, in Register_RELU6()
|
| D | logistic.cc | 31 namespace activations { namespace
139 return {/*init=*/activations::LogisticInit, in Register_LOGISTIC()
141 /*prepare=*/activations::LogisticPrepare, in Register_LOGISTIC()
142 /*invoke=*/activations::LogisticEval, in Register_LOGISTIC()
|
| D | tanh.cc | 32 namespace activations { namespace
147 return {/*init=*/activations::TanhInit, in Register_TANH()
149 /*prepare=*/activations::TanhPrepare, in Register_TANH()
150 /*invoke=*/activations::TanhEval, in Register_TANH()
|