Home
last modified time | relevance | path

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

/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.py30 self.assertIn("Add", api_def_map.op_names())
Dc_api_util.py137 def op_names(self): member in ApiDefMap
/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize_graph_test.py135 op_names = [op.name for op in g.get_operations()]
137 'scope1/post_activation_bypass_quant/' in name for name in op_names))
286 op_names = [op.name for op in graph.get_operations()]
287 self.assertTrue(any('test/Conv/weights_quant' in name for name in op_names))
288 self.assertTrue(any('test/Conv/act_quant' in name for name in op_names))
334 op_names = [op.name for op in graph.get_operations()]
337 any('scope1/Conv/act_quant' in name for name in op_names))
339 any('scope1/Conv/weights_quant' in name for name in op_names))
341 any('scope2/Conv/act_quant' in name for name in op_names))
343 any('scope2/Conv/weights_quant' in name for name in op_names))
[all …]
Dquantize_test.py328 op_names = [op.name for op in graph.get_operations()]
330 'test/test1/weights_quant/FakeQuantWithMinMaxVars' in op_names)
332 'test/test2/weights_quant/FakeQuantWithMinMaxVars' in op_names)
334 'test/test1/act_quant/FakeQuantWithMinMaxVars' in op_names)
335 self.assertTrue('test/act_quant/FakeQuantWithMinMaxVars' in op_names)
442 op_names = [op.name for op in graph.get_operations()]
444 'part/test/test/weights_quant/FakeQuantWithMinMaxVars' in op_names)
472 op_names = [op.name for op in graph.get_operations()]
474 'part/test/test/weights_quant/FakeQuantWithMinMaxVars' in op_names)
/external/tensorflow/tensorflow/python/grappler/
Ditem.i106 std::unordered_set<string> op_names; in TF_IdentifyImportantOps() local
108 op_names.insert(op->name()); in TF_IdentifyImportantOps()
111 op_names.insert(op->name()); in TF_IdentifyImportantOps()
118 if (op_names.find(node.name()) != op_names.end()) { in TF_IdentifyImportantOps()
129 for (const auto& op_name : op_names) { in TF_IdentifyImportantOps()
Dcluster_test.py135 op_names = gcluster.ListAvailableOps()
136 self.assertTrue('Add' in op_names)
137 self.assertTrue('MatMul' in op_names)
138 self.assertEqual(op_names, sorted(op_names))
Dcluster.i213 std::vector<string> op_names; in TF_ListAvailableOps() local
215 op_names.push_back(op.name()); in TF_ListAvailableOps()
217 std::sort(op_names.begin(), op_names.end()); in TF_ListAvailableOps()
220 PyObject* result = PyList_New(op_names.size()); in TF_ListAvailableOps()
221 for (int i = 0; i < op_names.size(); ++i) { in TF_ListAvailableOps()
222 PyList_SetItem(result, i, PyString_FromString(op_names[i].c_str())); in TF_ListAvailableOps()
/external/mesa3d/src/broadcom/qpu/
Dqpu_instr.c81 static const char *op_names[] = { in v3d_qpu_add_op_name() local
159 if (op >= ARRAY_SIZE(op_names)) in v3d_qpu_add_op_name()
162 return op_names[op]; in v3d_qpu_add_op_name()
168 static const char *op_names[] = { in v3d_qpu_mul_op_name() local
181 if (op >= ARRAY_SIZE(op_names)) in v3d_qpu_mul_op_name()
184 return op_names[op]; in v3d_qpu_mul_op_name()
/external/tensorflow/tensorflow/core/api_def/
Dapi_test.cc71 std::unordered_set<string> op_names; in TestAllApiDefsHaveCorrespondingOp() local
73 op_names.insert(op.name()); in TestAllApiDefsHaveCorrespondingOp()
76 ASSERT_TRUE(op_names.find(name_and_api_def.first) != op_names.end()) in TestAllApiDefsHaveCorrespondingOp()
/external/tensorflow/tensorflow/python/kernel_tests/
Dxent_op_test.py206 op_names = [
209 self.assertNotIn("BatchMatMul", op_names)
259 op_names = [
262 self.assertIn("BatchMatMul", op_names)
/external/tensorflow/tensorflow/core/kernels/
Dunary_ops_composition_test.cc35 void RunComposedOp(const std::vector<string> op_names, T input, T expected) { in RunComposedOp() argument
39 .Attr("op_names", op_names) in RunComposedOp()
Dunary_ops_composition.cc59 Status ExportComputeFns(const std::vector<string>& op_names, in ExportComputeFns()
61 for (const string& op_name : op_names) { in ExportComputeFns()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc573 string op_names; in Rewrite() local
575 strings::StrAppend(&op_names, nd->name(), ", "); in Rewrite()
578 << " to: " << op_names; in Rewrite()
720 const OpNameSet& op_names, in FindOpOccurrences() argument
723 for (const auto& it : op_names) { in FindOpOccurrences()
729 if (op_names.find(op_name) != op_names.end()) { in FindOpOccurrences()
Dscoped_allocator_optimizer.h59 void FindOpOccurrences(GraphDef* graph, const OpNameSet& op_names,
Darithmetic_optimizer.cc2922 std::vector<string> op_names = {root->op()}; in TrySimplify() local
2935 op_names.push_back(input.op()); in TrySimplify()
2945 if (op_names.size() == 1) return Status::OK(); in TrySimplify()
2952 std::reverse(op_names.begin(), op_names.end()); in TrySimplify()
2955 << str_util::Join(op_names, ", ") << "]"; in TrySimplify()
2965 SetAttrValue(op_names, &(*attr)["op_names"]); in TrySimplify()
Darithmetic_optimizer_test.cc3644 auto op_names = node.attr().at("op_names").list().s(); in TEST_F() local
3645 EXPECT_EQ(3, op_names.size()); in TEST_F()
3646 EXPECT_EQ("Sqrt", op_names[0]); in TEST_F()
3647 EXPECT_EQ("Log", op_names[1]); in TEST_F()
3648 EXPECT_EQ("Relu", op_names[2]); in TEST_F()
/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/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform.py438 op_names = _get_all_node_names_from_collection(
440 if op_names:
443 table_initializers = [t for t in op_names]
/external/python/pyasn1/pyasn1/type/
Dbase.py202 op_names = [name
210 for name in set(op_names):
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc269 std::vector<string> ImportExport(std::initializer_list<string> op_names) { in ImportExport() argument
272 AddOperatorsByName(op_names); in ImportExport()
274 for (const string& name : op_names) { in ImportExport()
/external/mesa3d/src/mesa/drivers/dri/r200/
Dr200_vertprog.c70 }op_names[]={ variable
339 for(i=0; i < sizeof(op_names) / sizeof(*op_names); i++) in op_operands()
340 if(op_names[i].opcode == opcode) in op_operands()
341 return op_names[i].ip; in op_operands()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc2346 auto op_names = {"ab", "abc", "abcd", "abcde"}; in TEST_F() local
2347 for (const auto& op_name : op_names) { in TEST_F()
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_ops.py1406 op_names = set([
1410 for op in op_names: