Searched refs:operand_multi_index (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1973 std::vector<llvm::Value*> operand_multi_index = source_index.multidim(); in EmitElementalConcatenate() local
1974 operand_multi_index[concat_dim] = b_->CreateNSWSub( in EmitElementalConcatenate()
1975 operand_multi_index[concat_dim], source_index_phis[operand_id]); in EmitElementalConcatenate()
1981 llvm_ir::IrArray::Index operand_index(operand_multi_index, operand->shape(), in EmitElementalConcatenate()
2115 std::vector<llvm::Value*> operand_multi_index; in EmitElementalGather() local
2126 operand_multi_index.push_back(index.GetConstantWithIndexType(0)); in EmitElementalGather()
2130 operand_multi_index.push_back(index[output_window_dim]); in EmitElementalGather()
2189 operand_multi_index[operand_dim] = in EmitElementalGather()
2190 Add(operand_multi_index[operand_dim], maybe_truncated_clamped_index); in EmitElementalGather()
2212 IrArray::Index operand_index(operand_multi_index, operand_shape, index_type); in EmitElementalGather()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 765 llvm::SmallVector<llvm::Value*> operand_multi_index(source_index.size()); in HandleSelectAndScatter() local
770 operand_multi_index[i] = in HandleSelectAndScatter()
774 ICmpULT(operand_multi_index[i], in HandleSelectAndScatter()
804 operand_multi_index, operand_array.GetShape(), b_.getInt64Ty()); in HandleSelectAndScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2016 std::vector<llvm::Value*> operand_multi_index(source_index.size()); in EmitSelectAndScatter() local
2030 operand_multi_index[i] = NSWSub(NSWAdd(strided_index, window_index[i]), in EmitSelectAndScatter()
2033 operand_multi_index[i], in EmitSelectAndScatter()
2059 IrArray::Index operand_index(operand_multi_index, operand_shape, in EmitSelectAndScatter()
|