Searched refs:EmitTuple (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.h | 68 void EmitTuple(const IrArray& tuple, absl::Span<llvm::Value* const> operands,
73 void EmitTuple(const IrArray& tuple, absl::Span<const IrArray> buffers,
|
| D | tuple_ops.cc | 72 void EmitTuple(const IrArray& tuple, absl::Span<llvm::Value* const> operands, in EmitTuple() function
84 void EmitTuple(const IrArray& tuple, absl::Span<const IrArray> buffers, in EmitTuple() function
91 llvm_ir::EmitTuple(tuple, buffer_ptrs, b); in EmitTuple()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 126 llvm_ir::EmitTuple(GetIrArray(hlo, hlo), target_arrays, &b_); in EmitTargetElementLoop()
|
| D | ir_emitter.cc | 147 llvm_ir::EmitTuple(GetIrArray(*tuple, *tuple), base_ptrs, &b_); in HandleTuple()
|
| D | ir_emitter_unnested.cc | 2204 llvm_ir::EmitTuple(GetIrArray(hlo, hlo), output_arrays, &b_); in EmitTargetElementLoopInThunk()
3098 llvm_ir::EmitTuple(GetIrArray(*unnested_hlo, *unnested_hlo), in EmitKernel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 344 llvm_ir::EmitTuple(GetIrArrayFor(infeed), {data_address, token_address}, &b_); in HandleInfeed()
374 llvm_ir::EmitTuple(llvm_ir::IrArray(data_address, data_shape), in HandleInfeed()
619 llvm_ir::EmitTuple(GetIrArrayFor(sort), destination_addresses, &b_); in HandleSort()
630 llvm_ir::EmitTuple(GetIrArrayFor(tuple), base_ptrs, &b_); in HandleTuple()
1346 llvm_ir::EmitTuple(GetIrArrayFor(crs), operand_ptrs, &b_); in HandleAllReduce()
2286 llvm_ir::EmitTuple(GetIrArrayFor(custom_call), base_ptrs, &b_); in HandleCustomCall()
2977 llvm_ir::EmitTuple(target_array, tuple_operand_ptrs, &b_); in EmitTargetElementLoop()
|