Searched defs:out_temp (Results 1 – 9 of 9) sorted by relevance
291 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()
902 const TensorShape& shape, Tensor* out_temp) TF_MUST_USE_RESULT { in forward_input_or_allocate_temp()
156 T out_temp = static_cast<T>(0); in convolve_nhwc() local214 T out_temp = static_cast<T>(0); in convolve_nhwc() local
712 Tensor* out_temp) { in allocate_scoped_tensor()722 const TensorShape& shape, Tensor* out_temp) { in forward_input_or_allocate_scoped_tensor()
295 Tensor* out_temp) { in allocate_scoped_tensor()305 const TensorShape& shape, Tensor* out_temp) { in forward_input_or_allocate_scoped_tensor()
90 at::Tensor out_temp = need_to_copy ? in nonzero_cuda_out_impl() local
135 T out_temp = static_cast<T>(0); in convolve_nchw() local
176 const float *out_temp = bias_is_considered ? output_nodes : layer_bias; in av1_nn_predict_avx2() local
155 Tensor* out_temp) { in allocate_temp()