Home
last modified time | relevance | path

Searched defs:op_def (Results 1 – 25 of 82) sorted by relevance

1234

/external/tensorflow/tensorflow/core/framework/
Dop_def_builder.h45 : op_def(def), shape_inference_fn(fn), is_function_op(is_function) {} in op_def() function
47 OpDef op_def; member
162 OpDef* op_def() { return &op_reg_data_.op_def; } in op_def() function
Dfake_input.cc59 FakeInputImpl::FakeInputImpl(const OpDef* op_def, int in_index, in FakeInputImpl()
201 return [dt](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
210 return [n](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
219 return [n, dt](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
232 return [dtv](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
233 NodeDefBuilder* builder) { in FakeInput()
Dop_def_util.cc150 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def) { in FindAttr()
159 OpDef::AttrDef* FindAttrMutable(StringPiece name, OpDef* op_def) { in FindAttrMutable()
168 const OpDef::ArgDef* FindInputArg(StringPiece name, const OpDef& op_def) { in FindInputArg()
194 static Status ValidateArg(const OpDef::ArgDef& arg, const OpDef& op_def, in ValidateArg()
251 Status ValidateOpDef(const OpDef& op_def) { in ValidateOpDef()
339 Status CheckOpDeprecation(const OpDef& op_def, int graph_def_version) { in CheckOpDeprecation()
389 string SummarizeOpDef(const OpDef& op_def) { in SummarizeOpDef()
488 void FillAttrMap(const OpDef& op_def, AttrMap* attr_map) { in FillAttrMap()
740 void RemoveNonDeprecationDescriptionsFromOpDef(OpDef* op_def) { in RemoveNonDeprecationDescriptionsFromOpDef()
754 void RemoveDescriptionsFromOpDef(OpDef* op_def) { in RemoveDescriptionsFromOpDef()
[all …]
Dnode_def_util.cc402 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypeForNode()
417 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypesForNode()
425 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypeForNode()
440 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypesForNode()
448 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InOutTypesForNode()
454 Status NumOutputsForNode(const NodeDef& node_def, const OpDef& op_def, in NumOutputsForNode()
462 Status ValidateNodeDef(const NodeDef& node_def, const OpDef& op_def) { in ValidateNodeDef()
559 const OpDef& op_def, int* num) { in ComputeArgRange()
580 const OpDef& op_def, NameRangeMap* result) { in NameRangesHelper()
593 Status NameRangesForNode(const NodeDef& node_def, const OpDef& op_def, in NameRangesForNode()
[all …]
Dnode_def_util_test.cc52 void ExpectSuccess(const NodeDef& good, const OpDef& op_def) { in ExpectSuccess()
58 void ExpectFailure(const NodeDef& bad, const OpDef& op_def, in ExpectFailure()
374 const OpDef op_def = ToOpDef(OpDefBuilder("Simple") in TEST() local
395 const OpDef op_def = ToOpDef(OpDefBuilder("Simple") in TEST() local
416 const OpDef op_def = ToOpDef(OpDefBuilder("Simple") in TEST() local
436 const OpDef op_def = ToOpDef(OpDefBuilder("Polymorphic") in TEST() local
462 const OpDef op_def = ToOpDef(OpDefBuilder("NRepeats") in TEST() local
502 const OpDef op_def = ToOpDef(OpDefBuilder("TypeList") in TEST() local
Dop_def_builder.cc146 void FinalizeAttr(StringPiece spec, OpDef* op_def, in FinalizeAttr()
336 void FinalizeInputOrOutput(StringPiece spec, bool is_output, OpDef* op_def, in FinalizeInputOrOutput()
425 void FinalizeControlOutput(StringPiece name, OpDef* op_def, in FinalizeControlOutput()
462 void FinalizeDoc(const string& text, OpDef* op_def, in FinalizeDoc()
639 OpDef* op_def = &op_reg_data->op_def; in Finalize() local
Dop_def_builder_test.cc48 OpDef& op_def = op_reg_data.op_def; in ExpectSuccess() local
68 OpDef& op_def = op_reg_data.op_def; in ExpectOrdered() local
632 const OpDef& op_def = op_reg_data.op_def; in TEST_F() local
Dop_registration_test.cc52 registry->SetWatcher([](const Status& s, const OpDef& op_def) -> Status { in TEST()
Dgraph_def_util.cc72 const OpDef* op_def; in AddDefaultAttrsToGraphDef() local
220 const OpDef* op_def; in StrippedOpListForGraph() local
Dnode_def_builder.h138 const OpDef& op_def() const { return *op_def_; } in op_def() function
Dgraph_def_util_test.cc31 Status FinalizeOpDef(const OpDefBuilder& b, OpDef* op_def) { in FinalizeOpDef()
231 OpDef* op_def = op_list.add_op(); in TEST() local
Dcommon_shape_fns_test.cc58 OpDef op_def = op_reg_data.op_def; in TEST() local
78 OpDef op_def = op_reg_data.op_def; in TEST() local
109 OpDef op_def = op_reg_data.op_def; in TEST() local
224 OpDef op_def = op_reg_data.op_def; in TEST() local
326 OpDef op_def = op_reg_data.op_def; in TEST() local
/external/toybox/toys/pending/
Dexpr.c135 static struct op_def { struct
154 void eval_op(struct op_def *o, struct value *ret, struct value *rhs) in eval_op() argument
/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_operation_table_test.cc31 bool HasResourceInputOrOutput(const OpDef& op_def) { in HasResourceInputOrOutput()
46 const OpDef* op_def; in TEST() local
Dxla_op_registry.cc43 const OpDef* op_def; in LaunchOpHasKernelForDevice() local
210 const OpDef* op_def; in RegisterCompilationKernels() local
361 const NodeDef& node_def, const OpKernel* op_kernel, const OpDef* op_def, in CompileTimeConstantInputs()
/external/tensorflow/tensorflow/js/ops/
Dts_op_gen.cc89 GenTypeScriptOp::GenTypeScriptOp(const OpDef& op_def, const ApiDef& api_def) in GenTypeScriptOp()
229 void WriteTSOp(const OpDef& op_def, const ApiDef& api_def, WritableFile* ts) { in WriteTSOp()
274 for (const auto& op_def : ops.op()) { in WriteTSOps() local
Dts_op_gen_test.cc146 const string op_def = R"( in TEST() local
182 const string op_def = R"( in TEST() local
/external/tensorflow/tensorflow/python/framework/
Dop_def_library.py37 def _Attr(op_def, name): argument
234 def __init__(self, op_def): argument
291 def add_op(self, op_def): argument
Dimporter.py138 def _FindAttrInOpDef(attr_name, op_def): argument
329 def _SetDefaultAttrValues(node_def, op_def): argument
Dauto_control_deps.py91 def op_is_stateful(op_def): argument
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.cc56 Status OpDefForOp(const char* op_name, const OpDef** op_def) { in OpDefForOp()
71 const OpDef* op_def = nullptr; in AttrTypeMapForOp() local
169 const OpDef* op_def = nullptr; in FillAttrValueMap() local
/external/tensorflow/tensorflow/java/src/gen/cc/
Dop_specs.cc46 explicit TypeResolver(const OpDef& op_def) : op_def_(op_def) {} in TypeResolver()
335 EndpointSpec CreateEndpoint(const OpDef& op_def, const ApiDef& api_def, in CreateEndpoint()
354 OpSpec OpSpec::Create(const OpDef& op_def, const ApiDef& api_def) { in Create()
/external/tensorflow/tensorflow/core/graph/
Dnode_builder.h128 const OpDef& op_def() const { return def_builder_.op_def(); } in op_def() function
Dgraph.cc46 const OpDef* op_def; // not owned member
171 const OpDef& Node::op_def() const { return *props_->op_def; } in op_def() function in tensorflow::Node
407 const OpDef* op_def; in AddNode() local
434 const OpDef* op_def; in CopyNode() local
/external/tensorflow/tensorflow/core/ops/
Dtpu_embedding_ops.cc88 auto* op_def = &op_reg_data->op_def; in RegisterPerTableLoadOpsForAlgorithmBody() local
205 auto* op_def = &op_reg_data->op_def; in RegisterPerTableRetrieveOpsForAlgorithmBody() local

1234