| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | approx_topk.cc | 42 std::vector<PrimitiveType> op_types; in GetOperandTypes() local
62 op_types.push_back(op_shape.element_type()); in GetOperandTypes()
64 return op_types; in GetOperandTypes()
71 XlaBuilder* builder, absl::Span<const PrimitiveType> op_types, in BuildReductionComputation() argument
73 auto num_operands = op_types.size(); in BuildReductionComputation()
80 for (const auto& op_type : op_types) { in BuildReductionComputation()
86 for (const auto& op_type : op_types) { in BuildReductionComputation()
123 auto op_types = status_or_optypes.value(); in AggregateToTopKBuilder() local
126 BuildReductionComputation(builder, op_types, comparator); in AggregateToTopKBuilder()
175 auto op_types = status_or_optypes.value(); in ApproxTopK() local
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | BUILD | 60 "//tensorflow/core/grappler:op_types",
99 "//tensorflow/core/grappler:op_types",
139 "//tensorflow/core/grappler:op_types",
174 "//tensorflow/core/grappler:op_types",
210 "//tensorflow/core/grappler:op_types",
246 "//tensorflow/core/grappler:op_types",
285 "//tensorflow/core/grappler:op_types",
326 "//tensorflow/core/grappler:op_types",
363 "//tensorflow/core/grappler:op_types",
490 "//tensorflow/core/grappler:op_types",
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | BUILD | 22 name = "op_types",
23 srcs = ["op_types.cc"],
24 hdrs = ["op_types.h"],
159 ":op_types",
178 ":op_types",
233 ":op_types",
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | BUILD | 44 "//tensorflow/core/grappler:op_types",
83 "//tensorflow/core/grappler:op_types",
120 "//tensorflow/core/grappler:op_types",
178 "//tensorflow/core/grappler:op_types",
207 "//tensorflow/core/grappler:op_types",
290 "//tensorflow/core/grappler:op_types",
362 "//tensorflow/core/grappler:op_types",
414 "//tensorflow/core/grappler:op_types",
458 "//tensorflow/core/grappler:op_types",
529 "//tensorflow/core/grappler:op_types",
[all …]
|
| /external/tensorflow/tensorflow/core/profiler/internal/advisor/ |
| D | operation_checker.h | 46 if (node->op_types().find("FusedBatchNorm") != node->op_types().end()) { in Check()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | BUILD | 32 "//tensorflow/core/grappler:op_types",
67 "//tensorflow/core/grappler:op_types",
97 "//tensorflow/core/grappler:op_types",
200 "//tensorflow/core/grappler:op_types",
308 "//tensorflow/core/grappler:op_types",
369 "//tensorflow/core/grappler:op_types",
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.h | 390 node_.mutable_op_types()->Reserve(op_types().size()); in ToProto()
391 for (const string& t : op_types()) { in ToProto()
443 op_types_.insert(node_.op_types().begin(), node_.op_types().end()); in FromProto()
688 const std::set<string>& op_types() const { return op_types_; } in op_types() function
800 op_types_.insert(node->op_types().begin(), node->op_types().end()); in SnapshotNodes()
854 const std::set<string>& op_types() const { return op_types_; } in op_types() function
874 for (const string& type : n.second->op_types()) { in pick_nodes()
|
| D | tfprof_show.cc | 129 for (const string& type : node->node->op_types()) { in ReAccount()
208 const std::set<string>& op_types = node->node->op_types(); in FormatNode() local
209 info.push_back(absl::StrJoin(op_types, "|")); in FormatNode()
|
| D | tfprof_op.cc | 297 std::set<string> op_types = node->node->op_types(); in FormatNode() local
298 attrs.push_back(absl::StrJoin(op_types, "|")); in FormatNode()
|
| D | tfprof_code.cc | 691 std::set<string> op_types = node->node->op_types(); in FormatNode() local
692 attrs.push_back(absl::StrJoin(op_types, "|")); in FormatNode()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cost_analyzer_test.py | 117 op_types = [b"MatMul", b"Conv2DBackpropFilter"]
122 op_types.append(b"Conv2D")
124 for op_type in op_types:
|
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | BUILD | 30 "//tensorflow/core/grappler:op_types",
66 "//tensorflow/core/grappler:op_types",
|
| /external/tensorflow/tensorflow/python/autograph/utils/ |
| D | testing.py | 146 def assertOpsNotCreated(self, op_types): argument
147 self.assertEmpty(set(op_types) & set(self.trace_log))
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | BUILD | 79 "//tensorflow/core/grappler:op_types",
284 "//tensorflow/core/grappler:op_types",
382 "//tensorflow/core/grappler:op_types",
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | dumping_callback_test.py | 917 op_types = [digest.op_type for digest in graph_op_digests]
918 self.assertIn("AddV2", op_types)
919 self.assertIn("Log", op_types)
920 self.assertIn("Sin", op_types)
1065 op_types = [digest.op_type for digest in graph_op_digests]
1066 self.assertIn("Less", op_types)
1067 self.assertIn("Mul", op_types)
1068 self.assertIn("AddV2", op_types)
|
| /external/parameter-framework/asio-1.10.6/include/asio/detail/ |
| D | select_reactor.hpp | 43 enum op_types { read_op = 0, write_op = 1, except_op = 2, enum in asio::detail::select_reactor
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_driver.cc | 522 std::vector<QuantParams> op_types; in GetBiasParams() local
523 op_types.reserve(non_biases.size()); in GetBiasParams()
526 op_types.push_back(non_bias_type.params); in GetBiasParams()
528 if (op_types.empty()) return {}; in GetBiasParams()
529 return func(op_types, legacy_float_scale_); in GetBiasParams()
|
| D | quantization_utils.cc | 469 const std::vector<quant::QuantizedType>& op_types, in GetUniformQuantizedTypeForBias() argument
471 if (op_types.empty()) return {}; in GetUniformQuantizedTypeForBias()
480 for (auto op_type : op_types) { in GetUniformQuantizedTypeForBias()
501 for (auto op_type : op_types) { in GetUniformQuantizedTypeForBias()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | op_callbacks_test.py | 433 op_types = []
441 op_types.append(compat.as_bytes(op_type))
452 self.assertIn(_ADD_OP, op_types)
453 self.assertIn(_LOG_OP, op_types)
|
| /external/pcre/src/sljit/ |
| D | sljitNativeX86_common.c | 1736 sljit_u32 op_types, argument
1742 sljit_u8 op_eax_imm = U8(op_types >> 24);
1743 sljit_u8 op_rm = U8((op_types >> 16) & 0xff);
1744 sljit_u8 op_mr = U8((op_types >> 8) & 0xff);
1745 sljit_u8 op_imm = U8(op_types & 0xff);
1843 sljit_u32 op_types, argument
1849 sljit_u8 op_eax_imm = U8(op_types >> 24);
1850 sljit_u8 op_rm = U8((op_types >> 16) & 0xff);
1851 sljit_u8 op_mr = U8((op_types >> 8) & 0xff);
1852 sljit_u8 op_imm = U8(op_types & 0xff);
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | analyzer_cli_test.py | 226 op_types = []
237 op_types.append(items[2])
253 sorted_op_types = sorted(op_types)
256 tst.assertEqual(sorted_op_types, op_types)
269 tst.assertEqual(op_type, op_types[index])
|
| /external/tensorflow/tensorflow/core/util/ |
| D | debug_events_writer_test.cc | 571 std::vector<string> op_types; in TEST_F() local
574 op_types.push_back(actuals[i].graph_op_creation().op_type()); in TEST_F()
579 EXPECT_EQ(op_types[i], "Log"); in TEST_F()
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | tfprof_log.proto | 85 repeated string op_types = 5; field
|
| /external/tensorflow/tensorflow/core/data/ |
| D | hash_utils.cc | 62 const std::array<const char*, SIZE>& op_types) { in IsNodeOfType() argument
63 for (const auto& type : op_types) { in IsNodeOfType()
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | BUILD | 389 "//tensorflow/core/grappler:op_types",
|