Home
last modified time | relevance | path

Searched refs:op_names (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen_main.cc74 std::vector<string> op_names = str_util::Split(arg, ','); in ParseOpListCommandLine() local
75 if (op_names.size() == 1 && op_names[0].empty()) { in ParseOpListCommandLine()
77 } else if (op_names.size() == 1 && op_names[0].substr(0, 1) == "@") { in ParseOpListCommandLine()
78 const string filename = op_names[0].substr(1); in ParseOpListCommandLine()
81 *op_list = std::move(op_names); in ParseOpListCommandLine()
Dc_api_util_test.py27 self.assertIn("Add", api_def_map.op_names())
/external/tensorflow/tensorflow/core/transforms/cse/
Dpass.cc51 DenseMap<Operation *, Attribute> op_names; in runOnOperation() local
54 op_names.insert({op, name}); in runOnOperation()
67 if (Attribute name = op_names.lookup(op)) { in runOnOperation()
/external/tensorflow/tensorflow/c/experimental/ops/gen/common/
Dpath_config.cc26 const std::vector<string> op_names) in PathConfig() argument
27 : output_path(output_dir), op_names(op_names) { in PathConfig()
Dpath_config.h25 std::vector<string> op_names; member
34 const std::vector<string> op_names);
Dcontroller.cc70 for (const auto& op_name : path_config_.op_names) { in BuildModel()
/external/tensorflow/tensorflow/python/grappler/
Ditem_wrapper.cc162 std::unordered_set<std::string> op_names; in PYBIND11_MODULE() local
164 op_names.insert(op->name()); in PYBIND11_MODULE()
167 op_names.insert(op->name()); in PYBIND11_MODULE()
174 if (op_names.find(node.name()) != op_names.end()) { in PYBIND11_MODULE()
184 for (const auto& op_name : op_names) { in PYBIND11_MODULE()
Dcluster_test.py131 op_names = gcluster.ListAvailableOps()
132 self.assertTrue('Add' in op_names)
133 self.assertTrue('MatMul' in op_names)
134 self.assertEqual(op_names, sorted(op_names))
Dcluster_wrapper.cc148 std::vector<std::string> op_names; in PYBIND11_MODULE() local
149 op_names.reserve(ops.size()); in PYBIND11_MODULE()
151 op_names.push_back(op.name()); in PYBIND11_MODULE()
153 std::sort(op_names.begin(), op_names.end()); in PYBIND11_MODULE()
154 return op_names; in PYBIND11_MODULE()
/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util.cc144 std::vector<std::string> op_names; in GetOperatorNames() local
145 op_names.reserve(model.operators.size()); in GetOperatorNames()
147 op_names.push_back(TryGetOperatorName(*op)); in GetOperatorNames()
149 return op_names; in GetOperatorNames()
234 const std::vector<std::string> op_names = GetOperatorNames(model); in PopulateConversionLog() local
235 for (const auto& op_name : op_names) { in PopulateConversionLog()
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dxent_op_test_base.py206 op_names = [
209 self.assertNotIn("BatchMatMul", op_names)
210 self.assertNotIn("BatchMatMulV2", op_names)
261 op_names = [
264 self.assertIn("BatchMatMulV2", op_names)
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dvariable_merger_pass.cc41 std::vector<string> op_names; in MergedOpFingerprint() local
42 op_names.reserve(ops.size()); in MergedOpFingerprint()
44 op_names.push_back(node->name()); in MergedOpFingerprint()
46 return Fingerprint64(absl::StrJoin(op_names, ",")); in MergedOpFingerprint()
/external/tensorflow/tensorflow/core/api_def/
Dapi_test.cc80 std::unordered_set<string> op_names; in TestAllApiDefsHaveCorrespondingOp() local
82 op_names.insert(op.name()); in TestAllApiDefsHaveCorrespondingOp()
85 ASSERT_TRUE(op_names.find(name_and_api_def.first) != op_names.end()) in TestAllApiDefsHaveCorrespondingOp()
/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
Dsparse_xent_op_test_base.py224 op_names = [
227 self.assertNotIn("BatchMatMul", op_names)
228 self.assertNotIn("BatchMatMulV2", op_names)
258 op_names = [
261 self.assertIn("BatchMatMulV2", op_names)
/external/tensorflow/tensorflow/python/kernel_tests/
Dxent_op_test_base.py210 op_names = [
213 self.assertNotIn("BatchMatMul", op_names)
214 self.assertNotIn("BatchMatMulV2", op_names)
265 op_names = [
268 self.assertIn("BatchMatMulV2", op_names)
/external/tensorflow/tensorflow/core/ir/
Dinterfaces_test.cc63 SmallVector<StringRef, 2> op_names = {"tfg.Foo", "tfg.Bar"}; in TEST() local
65 for (auto it : llvm::zip(op_names, expected)) { in TEST()
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_composition_test.cc55 void RunComposedOp(const std::vector<string> op_names, T input_scalar_value, in RunComposedOp() argument
66 .Attr("op_names", op_names) in RunComposedOp()
/external/mesa3d/src/broadcom/qpu/
Dqpu_instr.c82 static const char *op_names[] = { in v3d_qpu_add_op_name() local
166 if (op >= ARRAY_SIZE(op_names)) in v3d_qpu_add_op_name()
169 return op_names[op]; in v3d_qpu_add_op_name()
175 static const char *op_names[] = { in v3d_qpu_mul_op_name() local
188 if (op >= ARRAY_SIZE(op_names)) in v3d_qpu_mul_op_name()
191 return op_names[op]; in v3d_qpu_mul_op_name()
/external/tensorflow/tensorflow/core/kernels/
Dunary_ops_composition_test.cc36 void RunComposedOp(const std::vector<string> op_names, T input, T expected) { in RunComposedOp() argument
40 .Attr("op_names", op_names) in RunComposedOp()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc826 string op_names; in Rewrite() local
828 strings::StrAppend(&op_names, nd->name(), ", "); in Rewrite()
831 << " to: " << op_names; in Rewrite()
968 const OpNameSet& op_names, in FindOpOccurrences() argument
971 for (const auto& it : op_names) { in FindOpOccurrences()
977 if (op_names.find(op_name) != op_names.end()) { in FindOpOccurrences()
Dscoped_allocator_optimizer.h62 void FindOpOccurrences(GraphDef* graph, const OpNameSet& op_names,
/external/deqp/modules/gles31/scripts/
Dgen-implicit-conversions.py809 op_names = [
816 for name, desc, op in op_names:
827 op_names = [
836 for name, desc, op in op_names:
/external/python/pyasn1/pyasn1/type/
Dbase.py217 op_names = [name
225 for name in set(op_names):
/external/tensorflow/tensorflow/core/util/
Ddebug_events_writer_test.cc572 std::vector<string> op_names; in TEST_F() local
575 op_names.push_back(actuals[i].graph_op_creation().op_name()); in TEST_F()
577 std::sort(op_names.begin(), op_names.end()); in TEST_F()
580 EXPECT_EQ(op_names[i], strings::Printf("Log_%.2d", i * 3 + 2)); in TEST_F()
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_events_writer_test.py350 op_names = [trace.op_name for trace in reader.graph_execution_traces()]
351 self.assertLen(op_names, circular_buffer_size)
352 self.assertLen(op_names, len(set(op_names)))

12