Home
last modified time | relevance | path

Searched refs:EmitTuple (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dtuple_ops.h68 void EmitTuple(const IrArray& tuple, absl::Span<llvm::Value* const> operands,
79 void EmitTuple(const IrArray& tuple, absl::Span<const IrArray> buffers,
Dtuple_ops.cc61 void EmitTuple(const IrArray& tuple, absl::Span<llvm::Value* const> operands, in EmitTuple() function
74 void EmitTuple(const IrArray& tuple, absl::Span<const IrArray> buffers, in EmitTuple() function
81 llvm_ir::EmitTuple(tuple, buffer_ptrs, b); in EmitTuple()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.cc176 llvm_ir::EmitTuple(GetIrArray(hlo, hlo), target_arrays, &b_); in EmitTargetElementLoop()
Dir_emitter.cc220 llvm_ir::EmitTuple(GetIrArray(*tuple, *tuple), base_ptrs, &b_); in HandleTuple()
656 EmitTuple(tuple_array, allocas_for_returned_scalars, &b_); in ComputeNestedElement()
Dir_emitter_unnested.cc1855 llvm_ir::EmitTuple(*tuple_output_array, output_element_arrays, &b_); in EmitLoopFusionFromMlir()
4961 llvm_ir::EmitTuple(output_arrays.back(), in EmitHlo021Tile()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc408 llvm_ir::EmitTuple(GetIrArrayFor(infeed), {data_address, token_address}, &b_); in HandleInfeed()
438 llvm_ir::EmitTuple(llvm_ir::IrArray(data_address, data_shape), in HandleInfeed()
632 llvm_ir::EmitTuple(GetIrArrayFor(sort), destination_addresses, &b_); in HandleSort()
643 llvm_ir::EmitTuple(GetIrArrayFor(tuple), base_ptrs, &b_); in HandleTuple()
1097 llvm_ir::EmitTuple(GetIrArrayFor(crs), operand_ptrs, &b_); in HandleAllReduceSingleReplica()
1259 llvm_ir::EmitTuple(GetIrArrayFor(instruction), output_buffer_ptrs, &b_); in HandleAllToAll()
2252 llvm_ir::EmitTuple(GetIrArrayFor(hlo), tuple_operand_ptrs, &b_); in HandlePadToStatic()
2290 llvm_ir::EmitTuple(GetIrArrayFor(hlo), {out_values_ptr, out_indices_ptr}, in HandleTopK()
2346 llvm_ir::EmitTuple(GetIrArrayFor(custom_call), base_ptrs, &b_); in HandleCustomCall()
3161 llvm_ir::EmitTuple(target_array, tuple_operand_ptrs, &b_); in EmitTargetElementLoop()
[all …]