| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 190 op_def_.input_arg(arg_index).name()); in AddAttrForArg()
231 : op_def_.input_arg_size(); in FlattenInputs()
234 const auto& arg(op_def_.input_arg(i)); in FlattenInputs()
279 const auto& arg = *FindInputArg(api_def_.arg_order(i), op_def_); in Code()
292 for (int i = 0; i < op_def_.attr_size(); ++i) { in Code()
293 const auto& attr(op_def_.attr(i)); in Code()
335 attrs_.reserve(params_no_default_.size() - op_def_.input_arg_size() + in Code()
337 for (int i = op_def_.input_arg_size(), end = params_no_default_.size(); in Code()
403 param_names_[i + op_def_.input_arg_size()].GetRenameTo(); in Code()
407 for (int i = 0; i < op_def_.attr_size(); ++i) { in Code()
[all …]
|
| D | python_op_gen_internal.cc | 517 : op_def_(op_def), in GenPythonOp()
534 const auto& arg = *FindInputArg(api_def_.arg_order(i), op_def_); in Code()
563 attrs_.reserve(params_no_default.size() - op_def_.input_arg_size() + in Code()
565 for (int i = op_def_.input_arg_size(), end = params_no_default.size(); in Code()
689 const auto& arg = *FindInputArg(api_def_.arg_order(i), op_def_); in AddDocStringInputs()
697 ArgTypeName(op_def_, arg, inferred_attrs_, false)); in AddDocStringInputs()
708 const auto& attr = *FindAttr(name, op_def_); in AddDocStringAttrs()
799 strings::StrAppend(&prelude_, "_", AvoidPythonReserved(op_def_.name()), in AddOutputGlobals()
801 strings::StrAppend(&prelude_, " \"", AvoidPythonReserved(op_def_.name()), in AddOutputGlobals()
815 ArgTypeName(op_def_, op_def_.output_arg(i), inferred_attrs_, true)); in AddDocStringOutputs()
[all …]
|
| D | python_op_gen_internal.h | 98 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::__anon89074ceb0111::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 | 144 const OpDef& op_def() const { return *op_def_; } in op_def()
184 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()
105 status = ValidateNodeDef(node_def, op_def_); in ExpectInvalid()
114 OpDef op_def_; member in tensorflow::__anoneb56bbfe0111::NodeDefBuilderTest
|
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen.cc | 69 const OpDef& op_def_; member in tensorflow::__anonc3e3999b0111::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/compiler/mlir/tensorflow/c/ |
| D | c_api_unified_experimental_mlir.cc | 210 const tensorflow::OpDef* op_def_ = nullptr; member in mlir::TF::__anon2b880b700111::MlirAbstractOp
291 tensorflow::OpRegistry::Global()->LookUpOpDef(op, &op_def_)); in Reset()
292 assert(op_def_); in Reset()
341 if (current_ods_input_ != op_def_->input_arg_size()) in Create()
344 op_def_->input_arg_size(), " ; for op ", in Create()
349 for (const tensorflow::OpDef::ArgDef& output_arg : op_def_->output_arg()) { in Create()
572 if (current_ods_input_ >= op_def_->input_arg_size()) in AddInput()
575 op_def_->input_arg_size(), " allowed input_args ; for op ", in AddInput()
585 op_def_->input_arg(current_ods_input_++); in AddInput()
607 if (current_ods_input_ >= op_def_->input_arg_size()) in AddInputList()
[all …]
|
| /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::__anon373d94720111::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/core/common_runtime/eager/ |
| D | eager_operation.h | 95 const tensorflow::OpDef* OpDef() const override { return op_def_; }; in OpDef()
206 op_def_ = nullptr; in ClearInferenceState()
258 const tensorflow::OpDef* op_def_; // op definition from protobuf variable
|
| D | eager_operation.cc | 332 TF_RETURN_IF_ERROR(OpDefForOp(op, &op_def_)); in Reset()
354 if (!op_def_) return Status::OK(); in MaybeInferSingleInputAttrs()
356 const auto& input_def = op_def_->input_arg(inference_arg_idx_++); in MaybeInferSingleInputAttrs()
401 if (!op_def_) return Status::OK(); in InferInputListAttrs()
404 const auto& input_def = op_def_->input_arg(inference_arg_idx_++); in InferInputListAttrs()
|