| /external/tensorflow/tensorflow/core/graph/ |
| mkl_graph_util.h | 93 static inline bool IsMklOp(const std::string& op_name, DataType T) {
94 string kernel = KernelsRegisteredForOp(op_name);
107 static inline bool IsMklElementWiseOp(const std::string& op_name, DataType T) {
108 if (!IsMklOp(op_name, T)) {
111 bool result = (0 == op_name.compare(GetMklOpName("Add")) ||
112 0 == op_name.compare(GetMklOpName("Sub")) ||
113 0 == op_name.compare(GetMklOpName("Mul")) ||
114 0 == op_name.compare(GetMklOpName("Maximum")) ||
115 0 == op_name.compare(GetMklOpName("SquaredDifference")));
|
| graph_def_builder.cc | 101 Node* SourceOp(const string& op_name, const GraphDefBuilder::Options& opts) {
103 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name,
108 Node* UnaryOp(const string& op_name, NodeOut input,
111 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name,
117 Node* BinaryOp(const string& op_name, NodeOut a, NodeOut b,
120 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name,
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| remove_attribute.cc | 38 "argument, e.g. remove_attribute(op_name=Mul, attribute_name=foo)");
41 string op_name; local
42 if (context.params.count("op_name")) {
43 if (context.params.at("op_name").size() != 1) {
45 "remove_nodes expects a single op_name argument, but found ",
46 context.params.at("op_name").size());
48 op_name = context.params.at("op_name")[0];
50 op_name = "*";
58 if (((op_name == "*") || (op_name == node.op())) &
[all...] |
| rename_attribute.cc | 43 string op_name; local
44 if (context.params.count("op_name")) {
45 op_name = context.params.at("op_name")[0];
47 op_name = "*";
56 if (((op_name == "*") || (op_name == node.op())) &&
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function_testlib.h | 30 Output Call(Scope* scope, const string& op_name, const string& fn_name,
|
| session_state.cc | 73 const string& op_name = id.first.ToString(); local
74 auto it = tensors_.find(op_name);
77 string key = it->second.GetHandle(op_name);
|
| function_testlib.cc | 59 Output Call(Scope* scope, const string& op_name, const string& fn_name,
62 NodeDefBuilder builder(op_name, fn_name, scope->graph()->op_registry());
|
| /external/tensorflow/tensorflow/core/framework/ |
| dataset_stateful_op_whitelist.h | 27 Status Add(StringPiece op_name) {
28 op_names_.insert(op_name);
32 bool Contains(StringPiece op_name) {
33 return op_names_.find(op_name) != op_names_.end();
|
| op_registration_test.cc | 27 void Register(const string& op_name, OpRegistry* registry) {
28 registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status {
29 op_reg_data->op_def.set_name(op_name);
|
| kernel_def_builder.h | 34 explicit KernelDefBuilder(const char* op_name);
|
| /external/tensorflow/tensorflow/core/util/ |
| util.h | 24 // If op_name has '/' in it, then return everything before the first '/'.
26 StringPiece NodeNamePrefix(const StringPiece& op_name);
28 // If op_name has '/' in it, then return everything before the last '/'.
30 StringPiece NodeNameFullPrefix(const StringPiece& op_name);
|
| util.cc | 24 StringPiece NodeNamePrefix(const StringPiece& op_name) {
25 StringPiece sp(op_name);
34 StringPiece NodeNameFullPrefix(const StringPiece& op_name) {
35 StringPiece sp(op_name);
|
| /external/tensorflow/tensorflow/core/ops/ |
| image_ops_test.cc | 31 for (const char* op_name : {"ResizeArea", "ResizeBicubic", "ResizeBilinear",
33 ShapeInferenceTestOp op(op_name);
65 for (const char* op_name : {"DecodeJpeg", "DecodePng"}) {
66 ShapeInferenceTestOp op(op_name);
72 TF_ASSERT_OK(NodeDefBuilder("test", op_name)
78 TF_ASSERT_OK(NodeDefBuilder("test", op_name)
85 TF_ASSERT_OK(NodeDefBuilder("test", op_name)
94 const char* op_name = "DecodeAndCropJpeg"; local
95 ShapeInferenceTestOp op(op_name);
104 TF_ASSERT_OK(NodeDefBuilder("test", op_name)
128 const char* op_name = "DecodeAndCropJpeg"; local
[all...] |
| candidate_sampling_ops_test.cc | 25 for (const char* op_name : {
33 ShapeInferenceTestOp op(op_name);
|
| spectral_ops_test.cc | 23 for (const auto* op_name : {"FFT", "IFFT"}) {
24 ShapeInferenceTestOp op(op_name);
32 for (const auto* op_name : {"FFT2D", "IFFT2D"}) {
33 ShapeInferenceTestOp op(op_name);
41 for (const auto* op_name : {"FFT3D", "IFFT3D"}) {
42 ShapeInferenceTestOp op(op_name);
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| op_compatibility_lib.cc | 63 const string& op_name = op_list_.op(cur).name(); local
64 while (op_name > *iter) {
68 if (op_name == *iter) {
96 const string& op_name = op_list_.op(cur).name(); local
97 if (stable_ops_ != nullptr && stable_ops_->count(op_name) == 0) {
100 if (op_list_.op(cur).name() != op_name) break;
102 } else if (op_name < in_op_history.op(start).name()) {
109 } else if (op_name > in_op_history.op(start).name()) {
113 if (op_name <= in_op_history.op(start).name()) break;
126 if (in_op_history.op(end).name() != op_name) break
183 const string& op_name = op_list_.op(cur).name(); local
[all...] |
| /external/syslinux/com32/lua/src/ |
| dhcp.c | 104 static char op_name[64]; local
111 sprintf(op_name, "%u", option);
112 lua_pushstring(L, op_name);
118 sprintf(op_name, "%u.%d", option, done+1);
119 lua_pushstring(L, op_name);
150 static char op_name[64]; local
204 sprintf(op_name, "%u", option);
205 lua_pushstring(L, op_name);
214 sprintf(op_name, "%u", option);
215 lua_pushstring(L, op_name);
[all...] |
| /system/update_engine/scripts/update_payload/ |
| checker.py | 634 def _CheckLength(self, length, total_blocks, op_name, length_name):
640 op_name: Operation name (for error reporting).
648 raise error.PayloadError('%s: %s is zero.' % (op_name, length_name))
652 '%s: %s' % (op_name, length_name))
704 def _CheckReplaceOperation(self, op, data_length, total_dst_blocks, op_name):
711 op_name: Operation name for error reporting.
718 raise error.PayloadError('%s: contains src_extents.' % op_name)
722 raise error.PayloadError('%s: missing data_{offset,length}.' % op_name)
727 op_name + '.data_length', 'dst')
734 (op_name, data_length, total_dst_blocks, self.block_size)
[all...] |
| applier.py | 218 def _ApplyReplaceOperation(self, op, op_name, out_data, part_file, part_size):
223 op_name: name string for error reporting
242 '%s.dst_extents' % op_name):
277 (op_name, data_start, data_length))
279 def _ApplyMoveOperation(self, op, op_name, part_file):
288 op_name: name string for error reporting
301 '%s.dst_extents' % op_name)
303 def _ApplyZeroOperation(self, op, op_name, part_file):
308 op_name: name string for error reporting
315 base_name = '%s.dst_extents' % op_name
[all...] |
| block_tracer.py | 57 for op, op_name in common.OperationIter(operations, base_name,
64 op_name + '.dst_extents'):
87 op_name + '.src_extents'):
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| nn_test.py | 46 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/core/grappler/costs/ |
| analytical_cost_estimator.cc | 79 const string& op_name = op_context.name; local
83 inaccurate_nodes.push_back(op_name);
86 auto it = name_to_cost.find(op_name);
92 cost_node->set_name(op_name);
|
| /external/tensorflow/tensorflow/python/eager/ |
| execution_callbacks.py | 41 op_name,
50 op_name: Name of the op that generated the tensor with `inf`(s) or
58 self._op_name = op_name
90 def op_name(self): member in class:InfOrNanError
110 op_name,
130 op_name: Name of the TFE operation. This name is set by client and can be
176 error = InfOrNanError(op_type, op_name, index, len(outputs), value)
193 op_name,
201 op_name,
211 op_name,
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| human_readable_profile_builder.h | 45 void AddOp(tensorflow::StringPiece op_name,
51 {op_name.ToString(), short_name.ToString(), category.ToString(), cycles,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| hlo_metadata_test.cc | 60 EXPECT_EQ("my_sum_op", instruction->metadata().op_name());
85 EXPECT_EQ("", instruction->metadata().op_name());
|
