Searched refs:BuildKernelThunkForMlir (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 660 StatusOr<std::unique_ptr<KernelThunk>> BuildKernelThunkForMlir(
665 StatusOr<std::unique_ptr<KernelThunk>> BuildKernelThunkForMlir(
|
| D | ir_emitter_unnested.cc | 823 BuildKernelThunkForMlir(pad_to_static, mlir_input.thunk_info, in EmitPadToStaticFromMlir()
950 BuildKernelThunkForMlir(slice_to_dynamic, mlir_input.thunk_info, in EmitSliceToDynamicFromMlir()
1773 BuildKernelThunkForMlir(fusion, input.thunk_info, in EmitLoopFusionFromMlir()
1894 BuildKernelThunkForMlir(mlir_input.op, Thunk::ThunkInfo(), in HandleFusion()
1938 BuildKernelThunkForMlir(mlir_input.op, Thunk::ThunkInfo(), in HandleFusion()
2028 BuildKernelThunkForMlir(fusion_op, mlir_input.thunk_info, in HandleFusion()
2218 BuildKernelThunkForMlir( in EmitSelectAndScatterFromMlir()
2469 BuildKernelThunkForMlir(rng_op, rng_op.state(), mlir_input.thunk_info, in EmitRngGetAndUpdateState()
2526 BuildKernelThunkForMlir(scatter_op, scatter_op.getOperands().drop_front(), in EmitScatterFromMlir()
2992 BuildKernelThunkForMlir(sort_op, sort_op.output(), Thunk::ThunkInfo(), in EmitSortFromMlir()
[all …]
|