Searched refs:base_output_index (Results 1 – 2 of 2) sorted by relevance
294 const int64 base_output_index = (in_r * args.in_cols + in_c) * in_depth; in ComputeBackpropInput() local317 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); in ComputeBackpropInput()339 output[base_output_index + output_vectorized_size + j] = out_buf[j]; in ComputeBackpropInput()363 output[base_output_index + d] = accum; in ComputeBackpropInput()758 const int64 base_output_index = (out_r * args.out_cols + out_c) * out_depth; in ComputeBackpropFilter() local764 const bool fast_path = base_output_index <= output_last_vector_index; in ComputeBackpropFilter()772 Eigen::internal::ploadu<Packet>(out_backprop + base_output_index + i); in ComputeBackpropFilter()792 const int64 out_bprop_index = base_output_index + i; in ComputeBackpropFilter()
94 const int64 base_output_index = (out_r * args.out_cols + out_c) * out_depth; in Run() local115 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); in Run()134 base_output_index + output_vectorized_size; in Run()