Searched refs:x_iter_num (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 116 llvm::Value* x_loc, int64 x_iter_num)>;
260 int64 x_iter_num);
267 int64 x_iter_num);
274 int64 x_iter_num);
|
| D | ir_emitter_unnested.cc | 97 llvm::Value* x_loc, int64 x_iter_num)>;
2599 int GetPartialResultIndex(int64 x_iter_num) const { in GetPartialResultIndex()
2603 return x_iter_num; in GetPartialResultIndex()
2855 llvm::Value* x_loc, int64 x_iter_num) { in EmitTileElementForReduction() argument
2868 int partial_result_index = reduction_info->IsRowReduction() ? 0 : x_iter_num; in EmitTileElementForReduction()
3181 int64 x_iter_num) { in EmitKernel() argument
3183 unnested_hlo, index, kernel_info, y_loc, x_loc, x_iter_num); in EmitKernel()
3251 int64 x_iter_num) { in EmitHlo021Tile() argument
3252 EmitTileElementForCopy(hlo, index, kernel_info, y_loc, x_loc, x_iter_num); in EmitHlo021Tile()
3259 llvm::Value* x_loc, int64 x_iter_num) { in EmitHlo021Tile() argument
[all …]
|