Searched refs:input_mappings (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | einsum_spmd_expander.cc | 97 std::vector<absl::flat_hash_map<char, std::vector<int>>>& input_mappings, in ExtractEquationRelations() argument
132 input_mappings.resize(inputs.size()); in ExtractEquationRelations()
137 input_mappings[i][char_and_index.value()].emplace_back( in ExtractEquationRelations()
260 std::vector<absl::flat_hash_map<char, std::vector<int>>> input_mappings; in ComputeLayoutForward() local
264 input_mappings, output_mapping)); in ComputeLayoutForward()
265 if (input_mappings.size() != num_inputs) in ComputeLayoutForward()
267 "Einsum equation ", equation, " has ", input_mappings.size(), in ComputeLayoutForward()
292 /*replicate_incompatible_dimensions=*/true, layouts, input_mappings)); in ComputeLayoutForward()
315 std::vector<absl::flat_hash_map<char, std::vector<int>>> input_mappings; in ComputeLayoutBackward() local
319 input_mappings, output_mapping)); in ComputeLayoutBackward()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 647 llvm::SmallVector<llvm::SmallVector<int64_t, 4>, 4> input_mappings( in GetMetadataArgumentMapping() local
651 input_mappings.front().resize(metadata.args_size()); in GetMetadataArgumentMapping()
652 std::iota(input_mappings.front().begin(), input_mappings.front().end(), 0); in GetMetadataArgumentMapping()
653 return input_mappings; in GetMetadataArgumentMapping()
663 input_mappings[device].push_back(idx); in GetMetadataArgumentMapping()
665 for (auto& input : input_mappings) input.push_back(idx); in GetMetadataArgumentMapping()
668 input_mappings[sharding.tile_assignment_devices(0)].push_back(idx); in GetMetadataArgumentMapping()
672 return input_mappings; in GetMetadataArgumentMapping()
|
| /external/webrtc/audio/utility/ |
| D | channel_mixing_matrix.cc | 281 int input_mappings = 0; in CreateTransformationMatrix() local
285 if ((*matrix_)[output_ch][input_ch] != 1 || ++input_mappings > 1) in CreateTransformationMatrix()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_dynamic_layout_pass.cc | 211 llvm::SmallVector<llvm::SmallVector<int64_t, 4>, 4> input_mappings = in HandleCompileAndExecutes() local
219 llvm::zip(execute_launches, input_mappings)) { in HandleCompileAndExecutes()
|