Searched refs:EncodeArrayFunctionArguments (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_function.cc | 202 llvm::Value* EncodeArrayFunctionArguments( in EncodeArrayFunctionArguments() function
236 EncodeArrayFunctionArguments(parameter_addresses, name, b); in GetArrayFunctionCallArguments()
|
| H A D | ir_function.h | 135 llvm::Value* EncodeArrayFunctionArguments(
|
| H A D | ir_emitter.cc | 1240 EncodeArrayFunctionArguments(input_buffer_ptrs, "input_buffers", &b_); in HandleAllReduceMultipleReplica()
1242 EncodeArrayFunctionArguments(output_buffer_ptrs, "output_buffers", &b_); in HandleAllReduceMultipleReplica()
1317 EncodeArrayFunctionArguments(input_buffer_ptrs, "input_buffers", &b_); in HandleAllToAll()
1319 EncodeArrayFunctionArguments(output_buffer_ptrs, "output_buffers", &b_); in HandleAllToAll()
|