HomeSort by relevance Sort by last modified time
    Searched refs:output_vectorized_size (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
depthwise_conv_op.cc 91 const int64 output_vectorized_size = local
95 for (int i = 0; i < output_vectorized_size; i += kPacketSize) {
121 output_vectorized_size + j * padded_filter_inner_dim_size;
133 base_output_index + output_vectorized_size;
depthwise_conv_grad_op.cc 287 const int64 output_vectorized_size = (out_depth / kPacketSize) * kPacketSize; local
299 for (int i = 0; i < output_vectorized_size; i += kPacketSize) {
325 output_vectorized_size + j * padded_filter_inner_dim_size;
336 output[base_output_index + output_vectorized_size + j] = out_buf[j];
341 out_buffer[output_vectorized_size + j] = out_buf[j];
655 const int64 output_vectorized_size = local
    [all...]

Completed in 72 milliseconds