| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_ops_definitions.cc | 301 std::unordered_map<string, std::vector<DataTypeToOp>> op_map; in BuildOpNameToSocOpTypeMap() local
303 EmplaceOpType("INPUT", {}, SupportedOpType::INPUT, &op_map); in BuildOpNameToSocOpTypeMap()
304 EmplaceOpType("OUTPUT", {}, SupportedOpType::OUTPUT, &op_map); in BuildOpNameToSocOpTypeMap()
305 EmplaceOpType("NoOp", {}, SupportedOpType::NOP, &op_map); in BuildOpNameToSocOpTypeMap()
307 EmplaceOpType("FLATTEN", {}, SupportedOpType::FLATTEN, &op_map); in BuildOpNameToSocOpTypeMap()
310 EmplaceOpType("Identity", {}, SupportedOpType::NOP, &op_map); in BuildOpNameToSocOpTypeMap()
311 EmplaceOpType("Placeholder", {}, SupportedOpType::NOP, &op_map); in BuildOpNameToSocOpTypeMap()
312 EmplaceOpType("Const", {}, SupportedOpType::OP_CONST, &op_map); in BuildOpNameToSocOpTypeMap()
314 &op_map); in BuildOpNameToSocOpTypeMap()
316 &op_map); in BuildOpNameToSocOpTypeMap()
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | lift_to_graph.py | 63 def _copy_non_source(op, graph, op_map, base_graph): argument
89 copied_input = op_map.get(original_input, None)
107 copied_control_input = op_map.get(original_control_input, None)
136 op_map[op] = copied_op
138 op_map[o] = copied_op.outputs[i]
146 def _copy_source(s, graph, op_map, handle_captures, inverse_captures, argument
178 op=default_value.op, graph=graph, op_map=op_map,
187 input=op_map[default_value], shape=s.shape, name=s.op.name)
200 op_map[s] = copied_placeholder
203 op_map[s.op] = copied_placeholder.op
[all …]
|
| D | def_function.py | 1079 op_map = object_identity.ObjectIdentityDictionary()
1089 op_map = lift_to_graph.lift_to_graph(
1090 inits, ops.get_default_graph(), op_map=op_map)
1095 v.assign(op_map[init], read_value=False)
|
| /external/mesa3d/src/gallium/drivers/r600/sb/ |
| D | sb_gcm.cpp | 114 op_info &o = op_map[n]; in collect_instructions()
171 op_map[n].top_bb = bb; in td_schedule()
324 if (op_map[n].bottom_bb == bb) { in bu_sched_bb()
523 assert(op_map[n].bottom_bb == bu_bb); in bu_schedule()
654 op_info &oi = op_map[n]; in bu_release_op()
|
| D | sb_pass.h | 239 op_info_map op_map; variable
268 op_map(), uses(), nuc_stk(1), ucs_level(), in gcm()
|
| /external/mesa3d/src/gallium/drivers/vc4/ |
| D | vc4_state.c | 182 static const uint8_t op_map[] = { in tlb_stencil_setup_bits() local
196 bits |= op_map[state->zfail_op] << 25; in tlb_stencil_setup_bits()
197 bits |= op_map[state->zpass_op] << 22; in tlb_stencil_setup_bits()
198 bits |= op_map[state->fail_op] << 19; in tlb_stencil_setup_bits()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.pbtxt | 93 …_placeholders\', \'add_sources\', \'handle_captures\', \'base_graph\', \'op_map\'], varargs=None, …
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes.cc | 646 std::map<string, QuantizedOpInfo> op_map; in QuantizeNodes() local
651 op_map.insert({op_info.float_name, op_info}); in QuantizeNodes()
695 [&op_map, fallback_min, fallback_max, has_fallback_range]( in QuantizeNodes()
700 const QuantizedOpInfo& op_info = op_map[float_node.op()]; in QuantizeNodes()
|
| D | quantize_nodes_test.cc | 1222 std::map<string, int> op_map; in TestMergeDuplicateQuantizes() local
1224 ++op_map[node.op()]; in TestMergeDuplicateQuantizes()
1226 EXPECT_EQ(1, op_map["QuantizeV2"]); in TestMergeDuplicateQuantizes()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3344 std::map<std::string, std::pair<OpFunc, ValFunc>> op_map; in TEST_P() local
3347 op_map[name] = std::make_pair(CreateUnaryOp<op>, compute) in TEST_P()
3386 if (!op_map.count(op_name)) { in TEST_P()
3390 NodeDef node_def = op_map[op_name].first(tf_type_); in TEST_P()
3397 std::back_inserter(output_values), op_map[op_name].second); in TEST_P()
5624 std::map<std::string, std::pair<OpFunc, ValFunc>> op_map; in TEST_P() local
5626 op_map[name] = \ in TEST_P()
5635 op_map["Cast"] = std::make_pair(CreateCastOp, [](float x) { return x; }); in TEST_P()
5669 if (!op_map.count(op_name)) { in TEST_P()
5672 NodeDef node_def = op_map[op_name].first(tf_type_); in TEST_P()
[all …]
|
| /external/python/parse_type/tasks/_vendor/ |
| D | path.py | 1690 op_map = {
1695 return functools.partial(op_map[op], mask)
|
| /external/mesa3d/src/gallium/auxiliary/nir/ |
| D | nir_to_tgsi.c | 561 static enum tgsi_opcode op_map[][2] = { in ntt_emit_alu() enum
688 if (instr->op < ARRAY_SIZE(op_map) && op_map[instr->op][table_op64] != 0) { in ntt_emit_alu()
690 ureg_insn(c->ureg, op_map[instr->op][table_op64], in ntt_emit_alu()
|