Searched refs:extra_slice (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 72 absl::optional<MlirBufferSlice> extra_slice; member
662 Thunk::ThunkInfo thunk_info, absl::optional<MlirBufferSlice> extra_slice,
667 absl::optional<MlirBufferSlice> extra_slice,
|
| D | ir_emitter_unnested.cc | 824 mlir_input.extra_slice, &ir_arrays)); in EmitPadToStaticFromMlir()
951 mlir_input.extra_slice, &ir_arrays)); in EmitSliceToDynamicFromMlir()
1156 input.extra_slice->shape.tuple_shapes(1).dimensions(0); in EmitConvolutionThunkFromMlir()
1749 auto& slice = input.extra_slice.emplace(); in GetMlirEmitterInput()
1774 input.extra_slice, &ir_arrays)); in EmitLoopFusionFromMlir()
1895 mlir_input.extra_slice, &ir_arrays)); in HandleFusion()
1939 mlir_input.extra_slice, &ir_arrays)); in HandleFusion()
2029 mlir_input.extra_slice, &ir_arrays)); in HandleFusion()
2222 Thunk::ThunkInfo(), mlir_input.extra_slice, &ir_arrays)); in EmitSelectAndScatterFromMlir()
2470 mlir_input.extra_slice, &ir_arrays)); in EmitRngGetAndUpdateState()
[all …]
|