Searched refs:th_args (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_batch_kernels.cc | 50 MutableArrayRef<tfrt::TensorHandle> th_args, in ExecuteFallbackOp() argument
131 op(*exec_ctx_ptr, th_args, tfrt::OpAttrsRef(op_attrs), result_ths, op_chain); in ExecuteFallbackOp()
156 llvm::SmallVector<tfrt::TensorHandle, 8> th_args; in BatchFunctionFallback() local
157 th_args.reserve(args.size() + 1); in BatchFunctionFallback()
159 th_args.push_back(arg.CopyRef()); in BatchFunctionFallback()
170 th_args.emplace_back(host->GetHostDeviceRef(), dht->metadata(), in BatchFunctionFallback()
180 ExecuteFallbackOp(std::move(*op), th_args, &op_chain, results_refs, in BatchFunctionFallback()
|
| /external/ltp/testcases/kernel/mem/mtest05/ |
| D | mmstress.c | 285 long th_args[5]; /* argument list passed to thread_fault() */ in map_and_thread() local
372 th_args[1] = (long)map_addr; in map_and_thread()
373 th_args[2] = pagesize; in map_and_thread()
374 th_args[3] = fault_type; in map_and_thread()
376 th_args[0] = thrd_ndx; in map_and_thread()
377 th_args[4] = (long)0; in map_and_thread()
390 (void *)&th_args)) { in map_and_thread()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.cc | 1318 llvm::SmallVector<TensorHandle, 8> th_args; in Execute() local
1319 th_args.reserve(args_.size()); in Execute()
1334 th_args.push_back(down_cast<TensorHandleInterface*>(args_[i].get()) in Execute()
1341 if (th_args[i].IsMetadataAvailable()) { in Execute()
1342 auto arg_dtype = th_args[i].GetAvailableMetadata().dtype; in Execute()
1363 function_state_->GetFunc()(exec_ctx, th_args, OpAttrsRef(attrs_), in Execute()
1372 th_args.push_back( in Execute()
1382 for (auto& th_arg : th_args) { in Execute()
1388 (*op_)(exec_ctx, th_args, OpAttrsRef(attrs_), result_ths, chain); in Execute()
|