Home
last modified time | relevance | path

Searched defs:activations (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_gradient.py30 def get_gradients_through_compute_gradients(optimizer, loss, activations): argument
101 def hook_dummy_table_variables_to_activations(tpu_embedding, activations, argument
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drnn_common.py182 def select_last_activations(activations, sequence_lengths): argument
212 def mask_activations_and_labels(activations, labels, sequence_lengths): argument
248 def multi_value_predictions(activations, target_column, problem_type, argument
Ddynamic_rnn_estimator.py260 def _single_value_predictions(activations, argument
308 activations, labels, sequence_length, target_column, features): argument
331 activations, labels, sequence_length, target_column, features): argument
Ddynamic_rnn_estimator_test.py73 def get_eval_ops(self, features, activations, labels, metrics): argument
81 def loss(self, activations, labels, features): argument
Dstate_saving_rnn_estimator.py93 activations, labels, sequence_length, target_column, features): argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter_test.cc100 HloInstruction* activations = in TEST_F() local
141 HloInstruction* activations = in TEST_F() local
171 HloInstruction* activations = in TEST_F() local
201 HloInstruction* activations = in TEST_F() local
230 HloInstruction* activations = in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc135 auto activations = in Compile() local
Dconv_op_helpers.cc395 StringPiece type_string, xla::XlaOp activations, in MakeXlaBackpropFilterConvOp()
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dar_model.py339 def _predicted_covariance_op(self, activations, num_values): argument
360 def _predicted_mean_op(self, activations): argument
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dsoftmax.cc28 namespace activations { namespace
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_variants_test.cc1232 auto activations = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1258 auto activations = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1285 auto activations = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1317 auto activations = in XLA_TEST_F() local
1363 auto activations = ConstantLiteral(&builder, activations_literal); in XLA_TEST_F() local
/external/tensorflow/tensorflow/core/kernels/
Dlrn_op.cc327 auto activations = out_image.shaped<T, 2>({nodes * batch, depth}); in launch() local
Dmkl_lrn_op.cc564 auto activations = orig_output_tensor.shaped<T, 2>({nodes * batch, depth}); in MklDefaultToEigen() local
/external/tensorflow/tensorflow/lite/kernels/
Dactivations.cc38 namespace activations { namespace
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1627 std::vector<int64> activations = in TEST_F() local
1701 std::vector<int64> activations = in TEST_F() local
1806 std::vector<int64> activations = in TEST_F() local