Searched refs:operand_multi_index (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1771 std::vector<llvm::Value*> operand_multi_index = source_index.multidim(); in EmitElementalConcatenate() local
1772 operand_multi_index[concat_dim] = source_index_phis[operand_id]; in EmitElementalConcatenate()
1778 llvm_ir::IrArray::Index operand_index(operand_multi_index, operand->shape(), in EmitElementalConcatenate()
1877 std::vector<llvm::Value*> operand_multi_index; in EmitElementalGather() local
1886 operand_multi_index.push_back(index.GetConstantWithIndexType(0)); in EmitElementalGather()
1890 operand_multi_index.push_back(index[output_window_dim]); in EmitElementalGather()
1936 operand_multi_index[operand_dim] = in EmitElementalGather()
1937 Add(operand_multi_index[operand_dim], in EmitElementalGather()
1960 IrArray::Index operand_index(operand_multi_index, operand_shape, index_type); in EmitElementalGather()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 822 std::vector<llvm::Value*> operand_multi_index(source_index.size()); in HandleSelectAndScatter() local
827 operand_multi_index[i] = in HandleSelectAndScatter()
831 ICmpULT(operand_multi_index[i], in HandleSelectAndScatter()
858 operand_multi_index, operand_array.GetShape(), b_.getInt64Ty()); in HandleSelectAndScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 866 std::vector<llvm::Value*> operand_multi_index(source_index.size()); in HandleSelectAndScatter() local
871 operand_multi_index[i] = in HandleSelectAndScatter()
875 operand_multi_index[i], in HandleSelectAndScatter()
900 IrArray::Index operand_index(operand_multi_index, operand->shape(), in HandleSelectAndScatter()
|