Searched refs:operand_multi_index (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1818 std::vector<llvm::Value*> operand_multi_index = source_index.multidim(); in EmitElementalConcatenate() local
1819 operand_multi_index[concat_dim] = in EmitElementalConcatenate()
1820 NSWSub(operand_multi_index[concat_dim], source_index_phis[operand_id]); in EmitElementalConcatenate()
1826 llvm_ir::IrArray::Index operand_index(operand_multi_index, operand->shape(), in EmitElementalConcatenate()
1924 std::vector<llvm::Value*> operand_multi_index; in EmitElementalGather() local
1933 operand_multi_index.push_back(index.GetConstantWithIndexType(0)); in EmitElementalGather()
1937 operand_multi_index.push_back(index[output_window_dim]); in EmitElementalGather()
1996 operand_multi_index[operand_dim] = in EmitElementalGather()
1997 Add(operand_multi_index[operand_dim], maybe_truncated_clamped_index); in EmitElementalGather()
2019 IrArray::Index operand_index(operand_multi_index, operand_shape, index_type); in EmitElementalGather()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 741 std::vector<llvm::Value*> operand_multi_index(source_index.size()); in HandleSelectAndScatter() local
746 operand_multi_index[i] = in HandleSelectAndScatter()
750 ICmpULT(operand_multi_index[i], in HandleSelectAndScatter()
777 operand_multi_index, operand_array.GetShape(), b_.getInt64Ty()); in HandleSelectAndScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2304 std::vector<llvm::Value*> operand_multi_index(source_index.size()); in EmitSelectAndScatterFromMlir() local
2318 operand_multi_index[i] = NSWSub(NSWAdd(strided_index, window_index[i]), in EmitSelectAndScatterFromMlir()
2321 operand_multi_index[i], in EmitSelectAndScatterFromMlir()
2344 IrArray::Index operand_index(operand_multi_index, operand_shape, in EmitSelectAndScatterFromMlir()
|