| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 151 op_def_.input_arg(arg_index).name()); in AddAttrForArg()
190 : op_def_.input_arg_size(); in FlattenInputs()
193 const auto& arg(op_def_.input_arg(i)); in FlattenInputs()
238 const auto& arg = *FindInputArg(api_def_.arg_order(i), op_def_); in Code()
251 for (int i = 0; i < op_def_.attr_size(); ++i) { in Code()
252 const auto& attr(op_def_.attr(i)); in Code()
294 attrs_.reserve(params_no_default_.size() - op_def_.input_arg_size() + in Code()
296 for (int i = op_def_.input_arg_size(); i < params_no_default_.size(); ++i) { in Code()
327 param_names_[i + op_def_.input_arg_size()].GetRenameTo(); in Code()
331 for (int i = 0; i < op_def_.attr_size(); ++i) { in Code()
[all …]
|
| D | python_op_gen_internal.cc | 498 : op_def_(op_def), in GenPythonOp()
514 const auto& arg = *FindInputArg(api_def_.arg_order(i), op_def_); in Code()
543 attrs_.reserve(params_no_default.size() - op_def_.input_arg_size() + in Code()
545 for (int i = op_def_.input_arg_size(); i < params_no_default.size(); ++i) { in Code()
668 const auto& arg = *FindInputArg(api_def_.arg_order(i), op_def_); in AddDocStringInputs()
676 ArgTypeName(op_def_, arg, inferred_attrs_, false)); in AddDocStringInputs()
687 const auto& attr = *FindAttr(name, op_def_); in AddDocStringAttrs()
784 strings::StrAppend(&prelude_, "_", op_def_.name(), in AddOutputGlobals()
788 "\"", op_def_.name(), "\", ", lower_op_name_outputs, ")"); in AddOutputGlobals()
801 ArgTypeName(op_def_, op_def_.output_arg(i), inferred_attrs_, true)); in AddDocStringOutputs()
[all …]
|
| D | python_op_gen_internal.h | 96 const OpDef& op_def_; variable
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_builder.cc | 44 const Status status = op_registry->LookUpOpDef(string(op_name), &op_def_); in NodeDefBuilder()
61 : op_def_(op_def) { in NodeDefBuilder()
68 node_def_.set_op(op_def_->name()); in Initialize()
73 return &op_def_->input_arg(inputs_specified_++); in NextArgDef()
77 if (op_def_ == nullptr) { in NextArgAvailable()
79 } else if (inputs_specified_ >= op_def_->input_arg_size()) { in NextArgAvailable()
81 op_def_->input_arg_size(), in NextArgAvailable()
90 Status status = fake_input(*op_def_, inputs_specified_, node_def_, this); in Input()
217 if (op_def_ != nullptr && inputs_specified_ < op_def_->input_arg_size()) { in Finalize()
223 op_def_->input_arg_size(), " inputs in Op")); in Finalize()
[all …]
|
| D | fake_input.cc | 45 const OpDef* const op_def_; member in tensorflow::__anona398ba0d0111::FakeInputImpl
61 : op_def_(op_def), in FakeInputImpl()
151 const OpDef::AttrDef* attr = FindAttr(arg_->type_attr(), *op_def_); in GetDataType()
|
| D | node_def_builder.h | 138 const OpDef& op_def() const { return *op_def_; } in op_def()
173 const OpDef* op_def_; variable
|
| D | node_def_builder_test.cc | 38 op_def_ = op_reg_data.op_def; in Op()
44 EXPECT_FALSE(op_def_.name().empty()) << "Must call Op() before Builder()"; in Builder()
45 builder_.reset(new NodeDefBuilder("n", &op_def_)); in Builder()
73 status = ValidateNodeDef(node_def, op_def_); in ExpectSuccess()
103 status = ValidateNodeDef(node_def, op_def_); in ExpectInvalid()
112 OpDef op_def_; member in tensorflow::__anonc85c6fe00111::NodeDefBuilderTest
|
| D | op_kernel.cc | 111 NameRangesForNode(*def_, *context->op_def_, &input_name_map_, in OpKernel()
113 OP_REQUIRES_OK(context, CheckOpDeprecation(*context->op_def_, in OpKernel()
212 op_def_(op_def), in OpKernelConstruction()
|
| D | op_kernel.h | 381 const OpDef* op_def_; variable
|
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen.cc | 69 const OpDef& op_def_; member in tensorflow::__anon8001287d0111::GenTypeScriptOp
90 : op_def_(op_def), api_def_(api_def), num_outputs_(0) {} in GenTypeScriptOp()
109 auto op_def_arg = FindInputArg(api_def_.arg_order(i), op_def_); in ProcessArgs()
139 for (int i = 0; i < op_def_.attr_size(); i++) { in ProcessAttrs()
140 op_attrs_.push_back(OpAttrs(op_def_.attr(i), api_def_.attr(i))); in ProcessAttrs()
|
| /external/tensorflow/tensorflow/java/src/gen/cc/ |
| D | op_specs.cc | 46 explicit TypeResolver(const OpDef& op_def) : op_def_(op_def) {} in TypeResolver()
73 const OpDef op_def_; member in tensorflow::java::__anon9ce4e5540111::TypeResolver
103 for (const auto& attr_def : op_def_.attr()) { in TypeOf()
118 << "\" in operation \"" << op_def_.name() << "\""; in TypeOf()
165 << "\" in operation \"" << op_def_.name() << "\""; in TypesOf()
|
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| D | transform.py | 172 op_def_ = deepcopy(op.op_def)
176 [], input_types_, None, op_def_)
|