Searched refs:base_output_index (Results 1 – 2 of 2) sorted by relevance
306 const int64 base_output_index = (in_r * args.in_cols + in_c) * in_depth; in ComputeBackpropInput() local329 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); in ComputeBackpropInput()351 output[base_output_index + output_vectorized_size + j] = out_buf[j]; in ComputeBackpropInput()375 output[base_output_index + d] = accum; in ComputeBackpropInput()799 const int64 base_output_index = (out_r * args.out_cols + out_c) * out_depth; in ComputeBackpropFilter() local805 const bool fast_path = base_output_index <= output_last_vector_index; in ComputeBackpropFilter()813 Eigen::internal::ploadu<Packet>(out_backprop + base_output_index + i); in ComputeBackpropFilter()833 const int64 out_bprop_index = base_output_index + i; in ComputeBackpropFilter()
99 const int64 base_output_index = (out_r * args.out_cols + out_c) * out_depth; in Run() local120 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); in Run()139 base_output_index + output_vectorized_size; in Run()