Home
last modified time | relevance | path

Searched refs:output_mapping (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Deinsum_spmd_expander.cc98 absl::flat_hash_map<char, std::vector<int>>& output_mapping) { in ExtractEquationRelations() argument
115 if (output_mapping.contains(char_and_index.value())) in ExtractEquationRelations()
121 output_mapping[char_and_index.value()].emplace_back(char_and_index.index()); in ExtractEquationRelations()
261 absl::flat_hash_map<char, std::vector<int>> output_mapping; in ComputeLayoutForward() local
264 input_mappings, output_mapping)); in ComputeLayoutForward()
296 VerifyOrFixLayout(GetSpecsFromLabelsAndMap(output_mapping, in ComputeLayoutForward()
316 absl::flat_hash_map<char, std::vector<int>> output_mapping; in ComputeLayoutBackward() local
319 input_mappings, output_mapping)); in ComputeLayoutBackward()
330 {output_layout}, {output_mapping})); in ComputeLayoutBackward()
347 for (const auto& label_to_indices : output_mapping) in ComputeLayoutBackward()
[all …]
/external/pytorch/torch/fx/passes/utils/
Dfuser_utils.py158 output_mapping: Dict[Node, Node] = {} # mapping from old output to new outputs
164 output_mapping[node] = node_map[node]
167 outs = tuple(output_mapping.values())
184 original_outputs: Tuple[Node, ...] = tuple(output_mapping.keys())
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc357 std::vector<std::pair<int, int>>* output_mapping) { in RemoveFunctionOutputs() argument
358 DCHECK(output_mapping->empty()); in RemoveFunctionOutputs()
383 output_mapping->push_back({i, i - remove_output_args.size()}); in RemoveFunctionOutputs()
394 for (const auto& mapping : *output_mapping) { in RemoveFunctionOutputs()
Dfunctions.h157 std::vector<std::pair<int, int>>* output_mapping);
/external/mesa3d/src/gallium/frontends/d3d10umd/
DShaderParse.h173 unsigned *output_mapping);
DShader.cpp551 pShader->state.tokens = Shader_tgsi_translate(pCode, pShader->output_mapping); in CreateVertexShader()
687 pShader->state.tokens = Shader_tgsi_translate(pShaderCode, pShader->output_mapping); in CreateGeometryShader()
852 pShader->output_mapping); in CreateGeometryShaderWithStreamOutput()
1004 pShader->output_mapping); in CreatePixelShader()
1408 if (shader->output_mapping[i] == registerIndex) in ShaderFindOutputMapping()
DShaderTGSI.c1242 unsigned *output_mapping) in Shader_tgsi_translate() argument
2012 if (output_mapping) { in Shader_tgsi_translate()
2014 output_mapping[nr_outputs] in Shader_tgsi_translate()
2029 if (output_mapping) { in Shader_tgsi_translate()
2031 output_mapping[nr_outputs] in Shader_tgsi_translate()
2083 if (output_mapping) { in Shader_tgsi_translate()
2085 output_mapping[nr_outputs] in Shader_tgsi_translate()
DState.h60 unsigned output_mapping[PIPE_MAX_SHADER_OUTPUTS]; member
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc250 std::vector<std::pair<int, int>> output_mapping; member
331 for (const auto& pair : specialized_func.output_mapping) { in AddTensorMapping()
734 std::vector<std::pair<int, int>> output_mapping; in SpecializeFunction() local
743 TF_RETURN_IF_ERROR(RemoveFunctionOutputs(remove, &item, &output_mapping)); in SpecializeFunction()
772 control_deps, signature.active_outputs, output_mapping}; in SpecializeFunction()
/external/mesa3d/src/mesa/state_tracker/
Dst_program.c450 uint8_t output_mapping[VARYING_SLOT_TESS_MAX]; in st_translate_stream_output_info() local
451 memset(output_mapping, 0, sizeof(output_mapping)); in st_translate_stream_output_info()
461 output_mapping[attr] = num_outputs++; in st_translate_stream_output_info()
475 output_mapping[info->Outputs[i].OutputRegister]; in st_translate_stream_output_info()
/external/mesa3d/src/gallium/frontends/lavapipe/
Dlvp_pipeline.c557 uint8_t output_mapping[VARYING_SLOT_TESS_MAX]; in lvp_shader_xfb_init() local
558 memset(output_mapping, 0, sizeof(output_mapping)); in lvp_shader_xfb_init()
563 output_mapping[var->data.location + i] = var->data.driver_location + i; in lvp_shader_xfb_init()
575 … shader->stream_output.output[i].register_index = output_mapping[xfb_info->outputs[i].location]; in lvp_shader_xfb_init()