Searched defs:input_arg (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_builder.cc | 106 void NodeDefBuilder::SingleInput(const OpDef::ArgDef* input_arg, in SingleInput()
127 void NodeDefBuilder::ListInput(const OpDef::ArgDef* input_arg, in ListInput()
177 void NodeDefBuilder::VerifyInputType(const OpDef::ArgDef* input_arg, in VerifyInputType()
186 void NodeDefBuilder::VerifyInputRef(const OpDef::ArgDef* input_arg, in VerifyInputRef()
|
| D | node_def_builder.h | 165 DataType MaybeAddRef(const OpDef::ArgDef* input_arg, DataType dt) { in MaybeAddRef()
|
| D | op_def.proto | 52 repeated ArgDef input_arg = 2; field
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 583 auto input_arg = CreateParameterAndTransferLiteral(0, *input_literal, "p0", in DoIt() local
935 auto input_arg = CreateParameterAndTransferLiteral(0, *input_literal, "p0", in TEST_P() local
1040 auto input_arg = CreateParameterAndTransferLiteral(0, *input_literal, "p0", in DoIt() local
1224 auto input_arg = CreateParameterAndTransferLiteral(0, *input_literal, "p0", in TEST_P() local
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | update_api_def.cc | 44 for (auto& input_arg : op.input_arg()) { in FillBaseApiDef() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 438 def _create_hash_str(self, input_arg, output_arg, node_def): argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 1370 bool AddInputToOp(PyObject* input, const tensorflow::OpDef::ArgDef* input_arg,
1634 const auto& input_arg = op_def->input_arg(i); local
|