Searched refs:source_transform (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 246 auto source_transform = body_b.AddInstruction( in GetFinalFftUsingCollectivePermute() local
277 hlo->shape(), HloOpcode::kMultiply, phase_factor, source_transform)); in GetFinalFftUsingCollectivePermute()
295 source_transform = in GetFinalFftUsingCollectivePermute()
297 &body_b, source_transform, src_dst_pairs, (*next_channel_id)++); in GetFinalFftUsingCollectivePermute()
305 HloInstruction::CreateTuple({dest_transform, source_transform, in GetFinalFftUsingCollectivePermute()
|
| /external/executorch/examples/models/llava/ |
| D | export_llava.py | 109 .source_transform(source_transforms)
|
| /external/executorch/extension/llm/export/ |
| D | builder.py | 137 def source_transform( member in LLMEdgeManager
|
| /external/executorch/examples/models/llama/ |
| D | export_llama_lib.py | 564 .source_transform(_get_source_transforms(args.model, dtype_override, args))
|