Home
last modified time | relevance | path

Searched refs:op_name (Results 1 – 25 of 205) sorted by relevance

123456789

/external/tensorflow/tensorflow/core/graph/
Dmkl_graph_util.h94 static inline bool IsMklOp(const string& op_name, DataType T) { in IsMklOp() argument
95 string kernel = KernelsRegisteredForOp(op_name); in IsMklOp()
111 static inline bool IsMklOp(const string& op_name, DataType Tinput, in IsMklOp() argument
113 string kernel = KernelsRegisteredForOp(op_name); in IsMklOp()
129 static inline bool IsMklElementWiseOp(const string& op_name, DataType T) { in IsMklElementWiseOp() argument
130 if (!IsMklOp(op_name, T)) { in IsMklElementWiseOp()
133 bool result = (0 == op_name.compare(GetMklOpName("Add")) || in IsMklElementWiseOp()
134 0 == op_name.compare(GetMklOpName("Sub")) || in IsMklElementWiseOp()
135 0 == op_name.compare(GetMklOpName("Mul")) || in IsMklElementWiseOp()
136 0 == op_name.compare(GetMklOpName("Maximum")) || in IsMklElementWiseOp()
[all …]
Dgraph_def_builder.cc101 Node* SourceOp(const string& op_name, const GraphDefBuilder::Options& opts) { in SourceOp() argument
103 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in SourceOp()
108 Node* UnaryOp(const string& op_name, NodeOut input, in UnaryOp() argument
111 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in UnaryOp()
117 Node* BinaryOp(const string& op_name, NodeOut a, NodeOut b, in BinaryOp() argument
120 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in BinaryOp()
/external/tensorflow/tensorflow/core/ops/
Dimage_ops_test.cc31 for (const char* op_name : {"ResizeArea", "ResizeBicubic", "ResizeBilinear", in TEST()
33 ShapeInferenceTestOp op(op_name); in TEST()
65 for (const char* op_name : {"DecodeJpeg", "DecodePng"}) { in TEST()
66 ShapeInferenceTestOp op(op_name); in TEST()
72 TF_ASSERT_OK(NodeDefBuilder("test", op_name) in TEST()
78 TF_ASSERT_OK(NodeDefBuilder("test", op_name) in TEST()
85 TF_ASSERT_OK(NodeDefBuilder("test", op_name) in TEST()
94 const char* op_name = "DecodeAndCropJpeg"; in TEST() local
95 ShapeInferenceTestOp op(op_name); in TEST()
104 TF_ASSERT_OK(NodeDefBuilder("test", op_name) in TEST()
[all …]
Dspectral_ops_test.cc23 for (const auto* op_name : {"FFT", "IFFT"}) { in TEST()
24 ShapeInferenceTestOp op(op_name); in TEST()
32 for (const auto* op_name : {"FFT2D", "IFFT2D"}) { in TEST()
33 ShapeInferenceTestOp op(op_name); in TEST()
41 for (const auto* op_name : {"FFT3D", "IFFT3D"}) { in TEST()
42 ShapeInferenceTestOp op(op_name); in TEST()
/external/tensorflow/tensorflow/core/ops/compat/
Dop_compatibility_lib.cc63 const string& op_name = op_list_.op(cur).name(); in ValidateCompatible() local
64 while (op_name > *iter) { in ValidateCompatible()
68 if (op_name == *iter) { in ValidateCompatible()
96 const string& op_name = op_list_.op(cur).name(); in ValidateCompatible() local
97 if (stable_ops_ != nullptr && stable_ops_->count(op_name) == 0) { in ValidateCompatible()
100 if (op_list_.op(cur).name() != op_name) break; in ValidateCompatible()
102 } else if (op_name < in_op_history.op(start).name()) { in ValidateCompatible()
109 } else if (op_name > in_op_history.op(start).name()) { in ValidateCompatible()
113 if (op_name <= in_op_history.op(start).name()) break; in ValidateCompatible()
126 if (in_op_history.op(end).name() != op_name) break; in ValidateCompatible()
[all …]
/external/tensorflow/tensorflow/lite/experimental/writer/
Doption_writer_generator.cc195 for (auto op_name : ops_) { in BuildOpToOptionMap() local
199 ToCollapsed(op_name) + "options"; in BuildOpToOptionMap()
205 op_to_option_.insert(std::make_pair(op_name, option_name)); in BuildOpToOptionMap()
210 auto it = op_to_option_.find(op_name); in BuildOpToOptionMap()
212 std::cerr << "Didn't find option for " << op_name << std::endl; in BuildOpToOptionMap()
263 void GenerateImportForOp(FILE* fp, const std::string& op_name, in GenerateImportForOp() argument
274 fprintf(fp, " case BuiltinOperator_%s: {\n", op_name.c_str()); in GenerateImportForOp()
362 for (const auto& op_name : option->ops()) { in GenerateImport() local
363 auto option_it = option->op_to_option().find(op_name); in GenerateImport()
364 if (!option_it->second.empty() && ignores.find(op_name) == ignores.end()) in GenerateImport()
[all …]
/external/v8/src/s390/
Dassembler-s390.h609 #define DECLARE_S390_RIL_AB_INSTRUCTIONS(name, op_name, op_value) \ argument
612 ril_format(op_name, r1.code(), i2.immediate()); \
614 #define DECLARE_S390_RIL_C_INSTRUCTIONS(name, op_name, op_value) \ argument
616 ril_format(op_name, m1, i2.immediate()); \
632 #define DECLARE_S390_RR_INSTRUCTIONS(name, op_name, op_value) \ in S390_RIL_B_OPCODE_LIST() argument
634 rr_format(op_name, r1.code(), r2.code()); \ in S390_RIL_B_OPCODE_LIST()
637 rr_format(op_name, r1.code(), r2.code()); \
640 rr_format(op_name, m1, r2.code()); \
651 #define DECLARE_S390_RRD_INSTRUCTIONS(name, op_name, op_value) \ in S390_RR_OPCODE_LIST() argument
654 rrd_format(op_name, r1.code(), r3.code(), r2.code()); \ in S390_RR_OPCODE_LIST()
[all …]
/external/ltp/testcases/kernel/syscalls/statx/
Dstatx06.c104 char *op_name; member
107 .op_name = "Birth time"},
109 .op_name = "Modified time"},
111 .op_name = "Access time"},
113 .op_name = "Change time"}
153 tst_res(TFAIL, "%s < before time", tc->op_name); in test_statx()
155 tst_res(TFAIL, "%s > after_time", tc->op_name); in test_statx()
157 tst_res(TPASS, "%s Passed", tc->op_name); in test_statx()
/external/tensorflow/tensorflow/c/
Dkernels_test.cc70 const char* op_name, in GetFakeKernel() argument
73 def.set_op(op_name); in GetFakeKernel()
90 const char* op_name = "FooOp"; in TEST() local
93 REGISTER_OP(op_name) in TEST()
100 op_name, device_name, &MyCreateFunc, &MyComputeFunc, &MyDeleteFunc); in TEST()
106 TF_Buffer* buf = TF_GetRegisteredKernelsForOp(op_name, status); in TEST()
119 GetFakeKernel(device_name, op_name, &status); in TEST()
142 const char* op_name = "BarOp"; in TEST() local
145 REGISTER_OP(op_name) in TEST()
186 TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, nullptr, in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/
Dtrt_plugin_factory.cc44 PluginTensorRT* PluginFactoryTensorRT::CreatePlugin(const string& op_name) { in CreatePlugin() argument
45 if (!IsPlugin(op_name)) return nullptr; in CreatePlugin()
48 auto plugin_ptr = plugin_registry_[op_name].second(); in CreatePlugin()
55 const string& op_name, PluginDeserializeFunc deserialize_func, in RegisterPlugin() argument
57 if (IsPlugin(op_name)) return false; in RegisterPlugin()
61 op_name, std::make_pair(deserialize_func, construct_func)); in RegisterPlugin()
Dtrt_plugin_factory.h50 PluginTensorRT* CreatePlugin(const string& op_name);
52 bool RegisterPlugin(const string& op_name,
56 bool IsPlugin(const string& op_name) { in IsPlugin() argument
57 return plugin_registry_.find(op_name) != plugin_registry_.end(); in IsPlugin()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc47 bool HasOpName(const string& node_name, const string& op_name) { in HasOpName() argument
67 return node_name.substr(begin, end - begin) == op_name; in HasOpName()
412 const string& op_name, const string& sac_name, in BuildReplacementOp() argument
415 NodeDefBuilder op_builder(sa_op_name, op_name); in BuildReplacementOp()
441 const string& op_name, int sa_id, in BuildSplitNode() argument
472 const string& op_name, const string& sas_name) { in RewireSubgraph() argument
531 if (!HasOpName(n->name(), op_name)) { in RewireSubgraph()
535 << " " << op_name << " not in " << n->name(); in RewireSubgraph()
569 GraphDef* graph, const string& op_name, in Rewrite() argument
577 VLOG(1) << "UnaryElementwiseRewriter::Rewrite " << op_name in Rewrite()
[all …]
/external/tensorflow/tensorflow/core/grappler/
Dop_types.cc606 const string& op_name = node.op(); in IsStateful() local
607 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsStateful()
609 LOG(WARNING) << "Failed to lookup OpDef for " << op_name in IsStateful()
627 const string& op_name = node.op(); in IsFreeOfSideEffect() local
628 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsFreeOfSideEffect()
658 string op_name = node.op(); in ModifiesInputsInPlace() local
661 if (op_name == "AssignVariableOp" || op_name == "AssignAddVariableOp" || in ModifiesInputsInPlace()
662 op_name == "AssignSubVariableOp" || op_name == "ResourceScatterUpdate" || in ModifiesInputsInPlace()
663 op_name == "ResourceScatterAdd" || op_name == "ResourceScatterSub" || in ModifiesInputsInPlace()
664 op_name == "ResourceScatterMul" || op_name == "ResourceScatterDiv" || in ModifiesInputsInPlace()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dremove_attribute.cc40 string op_name; in RemoveAttribute() local
47 op_name = context.params.at("op_name")[0]; in RemoveAttribute()
49 op_name = "*"; in RemoveAttribute()
57 if (((op_name == "*") || (op_name == node.op())) && in RemoveAttribute()
Drename_attribute.cc42 string op_name; in RenameAttribute() local
44 op_name = context.params.at("op_name")[0]; in RenameAttribute()
46 op_name = "*"; in RenameAttribute()
55 if (((op_name == "*") || (op_name == node.op())) && in RenameAttribute()
/external/tensorflow/tensorflow/python/eager/
Dexecution_callbacks.py60 op_name, argument
77 self._op_name = op_name
109 def op_name(self): member in InfOrNanError
129 op_name, argument
193 error = InfOrNanError(op_type, op_name, index, len(outputs), value)
210 op_name, argument
218 op_name,
228 op_name, argument
236 op_name,
/external/tensorflow/tensorflow/python/framework/
Dc_api_util.py123 def get_api_def(self, op_name): argument
125 buf = c_api.TF_ApiDefMapGet(self._api_def_map, op_name, len(op_name))
132 def get_op_def(self, op_name): argument
133 if op_name in self._op_per_name:
134 return self._op_per_name[op_name]
135 raise ValueError("No entry found for " + op_name + ".")
/external/tensorflow/tensorflow/python/profiler/
Dtfprof_logger.py175 for op_name, entry in six.iteritems(logged_ops):
176 if op_name in all_ops:
177 all_ops[op_name].types.extend(entry.types)
178 if entry.float_ops > 0 and all_ops[op_name].float_ops == 0:
179 all_ops[op_name].float_ops = entry.float_ops
180 if entry.code_def.traces and not all_ops[op_name].code_def.traces:
181 all_ops[op_name].code_def.MergeFrom(entry.code_def)
183 all_ops[op_name] = entry
/external/tensorflow/tensorflow/python/grappler/
Dcost_analyzer.cc144 string op_name = op_perf.op().op(); in AnalyzeCosts() local
145 ops[op_name].count++; in AnalyzeCosts()
146 ops[op_name].time += op_perf.compute_cost(); in AnalyzeCosts()
147 ops[op_name].compute_time += op_perf.compute_time(); in AnalyzeCosts()
148 ops[op_name].memory_time += op_perf.memory_time(); in AnalyzeCosts()
149 ops[op_name].time_upper += op_perf.compute_time() + op_perf.memory_time(); in AnalyzeCosts()
150 ops[op_name].time_lower += in AnalyzeCosts()
152 ops[op_name].name = op_name; in AnalyzeCosts()
253 string op_name = perf.op().op(); in PrintAnalysis() local
254 os << std::setw(width) << op_name << ","; in PrintAnalysis()
/external/tensorflow/tensorflow/python/autograph/core/
Derrors.py57 def __init__(self, op_name, op_message, custom_traceback): argument
58 self.op_name = op_name
65 self.op_name)
213 op_name = error.op.name
215 rewritten_error = TfRuntimeError(op_name, op_message, cleaned_traceback)
/external/tensorflow/tensorflow/core/framework/
Ddataset_stateful_op_whitelist.h28 Status Add(string op_name) { in Add() argument
29 op_names_.insert(std::move(op_name)); in Add()
33 bool Contains(const string& op_name) { return op_names_.count(op_name); } in Contains() argument
Dgraph_def_util.cc129 static bool IsFunction(const GraphDef& graph_def, const string& op_name) { in IsFunction() argument
131 if (op_name == func_def.signature().name()) return true; in IsFunction()
204 for (const string& op_name : used_ops) { in OpsUsedByGraph() local
205 if (name_to_function.find(op_name) == name_to_function.end()) { in OpsUsedByGraph()
206 ops_used_in_graph->insert(op_name); in OpsUsedByGraph()
219 for (const string& op_name : used_ops) { in StrippedOpListForGraph() local
221 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(op_name, &op_def)); in StrippedOpListForGraph()
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dnn_test.py46 for op_name, tf_op, lt_op in ops:
50 self.assertIn(op_name, actual_lt.name)
65 for op_name, tf_op, lt_op in ops:
69 self.assertIn(op_name, actual_lt.name)
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc529 void AddOpConverter(const string& op_name, OpConverter op_converter) { in AddOpConverter() argument
530 converter_->op_registry_[op_name] = op_converter; in AddOpConverter()
2306 auto get_act_nodedef = [](string op_name) -> NodeDef { in TEST_F() argument
2309 if (op_name == "LeakyRelu") { in TEST_F()
2314 } else if (op_name == "Relu") { in TEST_F()
2317 } else if (op_name == "Relu6") { in TEST_F()
2320 } else if (op_name == "Sigmoid") { in TEST_F()
2323 } else if (op_name == "Tanh") { in TEST_F()
2331 auto get_act_output = [](string op_name, float input) -> float { in TEST_F() argument
2332 if (op_name == "LeakyRelu") { in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/util/
Dutil.cc24 StringPiece NodeNamePrefix(const StringPiece& op_name) { in NodeNamePrefix() argument
25 StringPiece sp(op_name); in NodeNamePrefix()
34 StringPiece NodeNameFullPrefix(const StringPiece& op_name) { in NodeNameFullPrefix() argument
35 StringPiece sp(op_name); in NodeNameFullPrefix()

123456789