| /external/tensorflow/tensorflow/python/distribute/failure_handling/ |
| D | gce_failure_handler_test.py | 107 input_arg, argument
247 def test_basic_run(self, input_arg, mwms_mode): argument
322 input_arg, mwms_mode): argument
397 def test_grace_period_continue_training(self, input_arg, mwms_mode): argument
|
| D | failure_handler_test.py | 119 input_arg='checkpoint', argument
230 def test_preemption_checkpointing(self, input_arg, mwms_mode): argument
353 def test_grace_period_continue_training(self, input_arg, mwms_mode): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_builder.cc | 115 void NodeDefBuilder::SingleInput(const OpDef::ArgDef* input_arg, in SingleInput()
136 void NodeDefBuilder::ListInput(const OpDef::ArgDef* input_arg, in ListInput()
186 void NodeDefBuilder::VerifyInputType(const OpDef::ArgDef* input_arg, in VerifyInputType()
195 void NodeDefBuilder::VerifyInputRef(const OpDef::ArgDef* input_arg, in VerifyInputRef()
|
| D | node_def_builder.h | 175 DataType MaybeAddRef(const OpDef::ArgDef* input_arg, DataType dt) { in MaybeAddRef()
|
| D | op_def.proto | 70 repeated ArgDef input_arg = 2; field
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 58 for (const InputArgInstantiation& input_arg : input_args_) { in GrapplerFunctionItem() local
327 const InputArgInstantiation& input_arg = item->input(input_index); in ReplaceInputWithConst() local
449 for (const InputArgInstantiation& input_arg : item.inputs()) { in Initialize() local
544 for (const InputArgInstantiation& input_arg : item.inputs()) { in MakeFunctionDef() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | noop_elimination.cc | 81 const string& input_arg) { in IsOutputIdentityOfInput()
|
| D | fusion_utils_test.cc | 39 for (const auto &input_arg : function.signature().input_arg()) in CheckUniqueNames() local
|
| D | slack_test.cc | 118 auto *input_arg = fdef->mutable_signature()->add_input_arg(); in TEST() local
|
| D | fusion_utils.cc | 161 for (const auto& input_arg : second_signature.input_arg()) { in GetUniqueSignature() local
387 for (const auto& input_arg : first_function.signature().input_arg()) { in LazyConjunctionNodes() local
|
| D | function_utils.cc | 107 auto* input_arg = fdef->mutable_signature()->mutable_input_arg()->Add(); in AddFunctionInput() local
|
| D | split_utils.cc | 96 for (const OpDef::ArgDef& input_arg : in InputRewriter() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_api_info.cc | 49 for (const auto& input_arg : function_def.signature().input_arg()) { in Init() local
|
| D | arithmetic_optimizer.cc | 2203 const OpDef::ArgDef& input_arg = op_def->input_arg(0); in SetInputType() local
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | update_api_def.cc | 46 for (auto& input_arg : op.input_arg()) { in FillBaseApiDef() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | op_def_library_pybind.cc | 435 const py::object& input_tensor, const ArgDef& input_arg, in GetInputType()
487 for (const ArgDef& input_arg : op_def.input_arg()) { in ExtractInputsAndAttrs() local
|
| D | function.py | 525 def _create_hash_str(self, input_arg, output_arg, node_def): argument
|
| /external/libaom/examples/ |
| D | noise_model.c | 81 static const arg_def_t input_arg = ARG_DEF("-i", "input", 1, "Input filename"); variable
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | tfrt_graph_execution_state.cc | 726 for (const auto& input_arg : op_def->input_arg()) { in EliminateRefVariablesFromV1ControlFlow() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 290 auto input_arg = src.cast<BlockArgument>(); in AddEdgeBetweenNodes() local
|
| D | import_model.cc | 624 auto& input_arg = signature.input_arg(i); in ConvertDeferredFunctions() local
|
| /external/cronet/base/ |
| D | command_line_unittest.cc | 334 const wchar_t* input_arg; in TEST() member
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 465 const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index]; in TF_GetNumberAttrForOpListInput() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 3498 const tensorflow::OpDef::ArgDef& input_arg,
3809 const auto& input_arg = op_def->input_arg(i); local
|