Searched refs:op_map (Results 1 – 7 of 7) sorted by relevance
/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 | 141 def _copy_non_source(op, graph, op_map): argument 152 copied_inputs = [op_map[x] for x in op.inputs] 153 copied_control_inputs = [op_map[x] for x in op.control_inputs] 161 op_map[op] = copied_op 163 op_map[o] = copied_op.outputs[i] 166 def _copy_source(s, graph, op_map, handle_captures, inverse_captures): argument 195 _copy_non_source(op=default_value.op, graph=graph, op_map=op_map) 199 input=op_map[default_value], shape=s.shape, name=s.op.name) 212 op_map[s] = copied_placeholder 289 op_map = {i: i for i in variable_init_tensors} # Pass through variables. [all …]
|
/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 | 181 static const uint8_t op_map[] = { in tlb_stencil_setup_bits() local 195 bits |= op_map[state->zfail_op] << 25; in tlb_stencil_setup_bits() 196 bits |= op_map[state->zpass_op] << 22; in tlb_stencil_setup_bits() 197 bits |= op_map[state->fail_op] << 19; in tlb_stencil_setup_bits()
|
/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()
|