Home
last modified time | relevance | path

Searched refs:use_function (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dgradient_checker.cc51 bool use_function) { in RunAndMaybeSum() argument
56 absl::MakeSpan(model_outputs), use_function)); in RunAndMaybeSum()
93 int input_index, bool use_function, in CalcNumericalGrad() argument
167 absl::MakeSpan(f_outputs), use_function)); in CalcNumericalGrad()
173 absl::MakeSpan(f_outputs), use_function)); in CalcNumericalGrad()
Dgradient_checker.h40 int input_index, bool use_function,
Dgradient_checker_test.cc36 float* expected_grad, int num_grad, bool use_function, in CompareNumericalAndManualGradients() argument
42 s = CalcNumericalGrad(ctx, model, inputs, input_index, use_function, in CompareNumericalAndManualGradients()
Dunified_api_testutil.h48 absl::Span<AbstractTensorHandle*> outputs, bool use_function);
Dunified_api_testutil.cc54 absl::Span<AbstractTensorHandle*> outputs, bool use_function) { in RunModel() argument
55 if (use_function) { in RunModel()
Dgradients_util.h81 absl::Span<AbstractTensorHandle*> outputs, bool use_function,
Dmnist_gradients_test.cc346 bool use_function = !std::get<2>(GetParam()); in TEST_P() local
347 if (use_function) { in TEST_P()
499 bool use_function = !std::get<2>(GetParam()); in TEST_P() local
500 if (use_function) { in TEST_P()
Dgradients_util.cc239 absl::Span<AbstractTensorHandle*> outputs, bool use_function, in RunModel() argument
241 if (use_function) { in RunModel()
/external/tensorflow/tensorflow/python/distribute/
Dmoving_averages_test.py51 distribution=all_distributions, mode=["eager"], use_function=[True, False])
150 def testUpdateContext(self, distribution, use_function): argument
192 def testReplicaContextEager(self, distribution, use_function): argument
193 if not use_function and isinstance(
216 if use_function:
224 def testCrossReplicaContextEager(self, distribution, use_function): argument
237 if use_function:
/external/tensorflow/tensorflow/c/experimental/gradients/
Dgrad_test_helper.cc27 absl::Span<AbstractTensorHandle* const> inputs, bool use_function, in CompareNumericalAndAutodiffGradients() argument
32 /*use_function=*/use_function); in CompareNumericalAndAutodiffGradients()
42 /*input_index=*/i, use_function, in CompareNumericalAndAutodiffGradients()
Dgrad_test_helper.h27 absl::Span<AbstractTensorHandle* const> inputs, bool use_function,
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util_test.cc287 for (const bool use_function : {false, true}) { in TEST()
290 if (use_function) { in TEST()
/external/tensorflow/tensorflow/python/eager/
Dbackprop_test.py1970 def test_degenerate_shape(self, use_function, use_pfor): argument
1978 if use_function:
/external/llvm/test/Bindings/OCaml/
Dcore.ml577 let fn = define_function "use_function" ty m in
/external/llvm-project/llvm/test/Bindings/OCaml/
Dcore.ml631 let fn = define_function "use_function" ty m in