Searched refs:window_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 382 const IrArray::Index window_index = loops.AddLoopsForShape( in MakeElementGenerator() local
384 CHECK_EQ(window_index.size(), index.size()); in MakeElementGenerator()
395 NSWMul(window_index[i], in MakeElementGenerator()
|
| D | ir_emitter_unnested.cc | 858 const IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter() local
872 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 1995 DimensionVector window_index(window.dimensions_size()); in HandleReduceWindow()
2005 std::fill(window_index.begin(), window_index.end(), 0); in HandleReduceWindow()
2828 DimensionVector window_index(rank); in IterateThroughWindow()
2829 std::fill(window_index.begin(), window_index.end(), 0); in IterateThroughWindow()
2836 window_index[i] * window.dimensions(i).window_dilation() - in IterateThroughWindow()
2854 IndexUtil::BumpIndices(window_shape, absl::MakeSpan(window_index))); in IterateThroughWindow()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 662 const llvm_ir::IrArray::Index window_index = loops.AddLoopsForShape( in EmitElementalReduceWindow() local
664 CHECK_EQ(window_index.size(), index.size()); in EmitElementalReduceWindow()
675 NSWMul(window_index[i], in EmitElementalReduceWindow()
815 const llvm_ir::IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter() local
828 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()
|