Searched refs:input_mappings (Results 1 – 3 of 3) sorted by relevance
617 llvm::SmallVector<llvm::SmallVector<int64_t, 4>, 4> input_mappings( in GetMetadataArgumentMapping() local621 input_mappings.front().resize(metadata.args_size()); in GetMetadataArgumentMapping()622 std::iota(input_mappings.front().begin(), input_mappings.front().end(), 0); in GetMetadataArgumentMapping()623 return input_mappings; in GetMetadataArgumentMapping()633 input_mappings[device].push_back(idx); in GetMetadataArgumentMapping()635 for (auto& input : input_mappings) input.push_back(idx); in GetMetadataArgumentMapping()638 input_mappings[sharding.tile_assignment_devices(0)].push_back(idx); in GetMetadataArgumentMapping()642 return input_mappings; in GetMetadataArgumentMapping()
281 int input_mappings = 0; in CreateTransformationMatrix() local285 if ((*matrix_)[output_ch][input_ch] != 1 || ++input_mappings > 1) in CreateTransformationMatrix()
228 llvm::SmallVector<llvm::SmallVector<int64_t, 4>, 4> input_mappings = in HandleCompileAndExecutes() local236 llvm::zip(execute_launches, input_mappings)) { in HandleCompileAndExecutes()