Home
last modified time | relevance | path

Searched defs:node_def (Results 1 – 25 of 134) sorted by relevance

123456

/external/tensorflow/tensorflow/python/framework/
Derrors_impl.py70 def __init__(self, node_def, op, message, error_code): argument
119 def node_def(self): member in OpError
231 def __init__(self, node_def, op, message): argument
250 def __init__(self, node_def, op, message, error_code=UNKNOWN): argument
270 def __init__(self, node_def, op, message): argument
285 def __init__(self, node_def, op, message): argument
303 def __init__(self, node_def, op, message): argument
320 def __init__(self, node_def, op, message): argument
338 def __init__(self, node_def, op, message): argument
353 def __init__(self, node_def, op, message): argument
[all …]
Ddevice.py126 def __call__(self, node_def): argument
135 def shortcut_string_merge(self, node_def): argument
Dfunction_def_to_graph.py266 def _get_num_args(arg_def, node_def): argument
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util_test.cc41 NodeDef node_def; in ToNodeDef() local
47 NodeDef node_def; in ToNodeDef() local
79 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
144 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
164 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
190 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
220 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
279 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
415 const NodeDef node_def = ToNodeDef(std::move( in TEST() local
436 const NodeDef node_def = ToNodeDef(std::move( in TEST() local
[all …]
Dnode_def_util.cc53 AttrSlice::AttrSlice(const NodeDef& node_def) in AttrSlice()
103 string SummarizeNodeDef(const NodeDef& node_def, int max_inputs_in_summary) { in SummarizeNodeDef()
124 string SummarizeAttrs(const NodeDef& node_def) { in SummarizeAttrs()
138 string FormatNodeDefForError(const NodeDef& node_def) { in FormatNodeDefForError()
326 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name) { in HasNodeAttr()
511 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypeForNode()
527 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypesForNode()
535 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypeForNode()
551 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypesForNode()
567 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InOutTypesForNode()
[all …]
Dfake_input.cc60 const NodeDef* node_def, NodeDefBuilder* builder) 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()
Dnode_properties_test.cc62 const NodeDef& node_def, in ValidateNodeProperties()
83 NodeDef node_def; in TEST() local
105 NodeDef node_def; in TEST() local
Dgraph_def_util.cc72 NodeDef* node_def = graph_def->mutable_node(i); in AddDefaultAttrsToGraphDef() local
86 NodeDef* node_def, const OpRegistryInterface& consumer_op_registry, in RemoveNewDefaultAttrsFromNodeDef()
144 NodeDef* node_def = graph_def->mutable_node(n); in RemoveNewDefaultAttrsFromGraphDef() local
154 NodeDef* node_def = func_def->mutable_node_def(n); in RemoveNewDefaultAttrsFromGraphDef() local
Dnode_properties.cc25 NodeDef node_def, const OpRegistryInterface* op_registry, in CreateFromNodeDef()
Dmemory_types_test.cc58 NodeDef node_def; in TEST() local
Dfunction.proto94 repeated NodeDef node_def = 3; field
Dop_kernel_test.cc148 NodeDef node_def; in CreateNodeDef() local
181 NodeDef node_def; in ExpectFailure() local
327 NodeDef node_def = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F() local
336 NodeDef node_def = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F() local
484 NodeDef node_def; in CreateNodeDef() local
1006 const NodeDef& node_def, const char* input_name, in BM_InputRangeHelper()
1031 NodeDef node_def; in BM_ConcatInputRange() local
1052 NodeDef node_def; in BM_SelectInputRange() local
1069 NodeDef node_def; in BM_TraceString() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc155 NodeDef node_def; local
765 Status GetInputs(const NodeDef& node_def, in GetInputs()
804 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F() local
830 NodeDef node_def; in TEST_F() local
887 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F() local
1639 void RunValidationAndConversion(const NodeDef& node_def, in RunValidationAndConversion()
1670 const NodeDef& node_def, const Status& status, const char* output_name, in RunValidationAndConversion()
1911 const string& name, const NodeDef node_def, in TestOpConverterMultiOut()
1927 void TestOpConverter(const string& name, const NodeDef node_def, in TestOpConverter()
2015 NodeDef node_def; in TestConvertConst() local
[all …]
Dconvert_nodes.cc1102 const NodeDef& node_def, const std::vector<TRT_TensorOrWeights>& inputs, in OpConverterParams()
1116 Converter* converter, const NodeDef& node_def, in OpConverterParams()
1148 const NodeDef& node_def, int output_port, in ConvertToTensorOrWeights()
1286 Status Converter::ConvertNode(const NodeDef& node_def) { in ConvertNode()
1583 const NodeDef& node_def, in TransposeTensor()
1671 void Converter::SetLayerName(nvinfer1::ILayer* layer, const NodeDef& node_def, in SetLayerName()
1707 const NodeDef& node_def, in PrepareTensorForShape()
2005 Status Converter::GetInputs(const NodeDef& node_def, in GetInputs()
2054 const auto& node_def = params.node_def; in CheckInputsWeights() local
2098 Status GetNodeDefTfType(const NodeDef& node_def, DataType* tf_type, in GetNodeDefTfType()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor.cc60 inline bool IsMerge(const NodeDef& node_def) { in IsMerge()
65 inline bool IsNextIteration(const NodeDef& node_def) { in IsNextIteration()
648 const NodeDef& node_def = get_node_def(n); in BuildNodeIndex() local
696 const NodeDef& node_def = get_node_def(n); in InitFromEdges() local
704 const NodeDef& node_def = get_node_def(n); in InitFromEdges() local
758 const NodeDef& node_def) { in ValidateColocationConstraints()
775 Status GraphConstructor::MakeNode(NodeDef&& node_def, Node** node) { in MakeNode()
835 Status GraphConstructor::ModifyNodeDefForImport(NodeDef* node_def) { in ModifyNodeDefForImport()
846 void RemoveInputs(const std::vector<int>& inputs_to_remove, NodeDef* node_def, in RemoveInputs()
868 NodeDef* node_def, std::vector<bool>* input_already_exists) { in RemapNodeDefInputs()
[all …]
/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util_test.cc43 tensorflow::NodeDef node_def; in TEST() local
93 tensorflow::NodeDef node_def; in TEST() local
191 tensorflow::NodeDef node_def; in TEST() local
/external/tensorflow/tensorflow/core/util/
Dmirror_pad_mode.cc24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name, in GetNodeAttr()
/external/tensorflow/tensorflow/python/util/
Dkernel_registry.cc29 tensorflow::NodeDef node_def; in TryFindKernelClass() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc89 const NodeDef& node_def, int num_cores_per_replica, bool add_metadata) { in ParseShardingFromDevice()
151 const NodeDef& node_def, bool add_metadata) { in GetShardingFromNodeDef()
/external/tensorflow/tensorflow/python/compiler/xla/
Djit.py123 def xla_compile(node_def): argument
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.cc131 void OpsTestBase::set_node_def(const NodeDef& node_def) { in set_node_def()
135 NodeDef* OpsTestBase::node_def() { return &node_def_; } in node_def() function in tensorflow::OpsTestBase
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dvariable_merger_pass.cc75 NodeDef node_def; in MergeVarHandleOps() local
105 NodeDef node_def; in MergeReadVariableOps() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dtfr_decompose_ctx.cc121 StatusOr<FunctionDef> TFRDecomposeContext::ExpandNode(const NodeDef& node_def, in ExpandNode()
209 StatusOr<FunctionDef> ExpandNode(const NodeDef& node_def, in ExpandNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dvectorizer_registry_test.cc43 NodeDef node_def; in TEST() local
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_deserialization.py428 def _check_op_has_custom_gradients(node_def): argument
434 def fix_node_def(node_def, functions, shared_name_suffix): argument

123456