Home
last modified time | relevance | path

Searched refs:op_map (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_ops_definitions.cc301 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/
Dlift_to_graph.py141 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/
Dsb_gcm.cpp114 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()
Dsb_pass.h239 op_info_map op_map; variable
268 op_map(), uses(), nuc_stk(1), ucs_level(), in gcm()
/external/mesa3d/src/gallium/drivers/vc4/
Dvc4_state.c181 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/
Dquantize_nodes.cc646 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()
Dquantize_nodes_test.cc1222 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()