Home
last modified time | relevance | path

Searched defs:out_temp (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.cc291 Tensor* out_temp) { in allocate_temp()
310 Tensor* out_temp, in allocate_temp()
582 Tensor* out_temp) { in forward_input_or_allocate_temp()
821 DataType type, const TensorShape& shape, Tensor* out_temp, in allocate_temp()
859 Tensor* out_temp, in allocate_temp()
866 Tensor* out_temp) { in allocate_temp()
H A Dop_kernel.h902 const TensorShape& shape, Tensor* out_temp) TF_MUST_USE_RESULT { in forward_input_or_allocate_temp()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/directconv2d/nhwc/neon/
H A Dimpl.cpp156 T out_temp = static_cast<T>(0); in convolve_nhwc() local
214 T out_temp = static_cast<T>(0); in convolve_nhwc() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Drocm_solvers.cc712 Tensor* out_temp) { in allocate_scoped_tensor()
722 const TensorShape& shape, Tensor* out_temp) { in forward_input_or_allocate_scoped_tensor()
H A Dcuda_solvers.cc295 Tensor* out_temp) { in allocate_scoped_tensor()
305 const TensorShape& shape, Tensor* out_temp) { in forward_input_or_allocate_scoped_tensor()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNonzero.cu90 at::Tensor out_temp = need_to_copy ? in nonzero_cuda_out_impl() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/directconv2d/nchw/
H A Dall.cpp135 T out_temp = static_cast<T>(0); in convolve_nchw() local
/aosp_15_r20/external/libaom/av1/encoder/x86/
H A Dml_avx2.c176 const float *out_temp = bias_is_considered ? output_nodes : layer_bias; in av1_nn_predict_avx2() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dtfrt_op_kernel.cc155 Tensor* out_temp) { in allocate_temp()