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

  /external/tensorflow/tensorflow/core/kernels/
conv_grad_input_ops.cc 413 const int filter_total_size = dims.spatial_dims[0].filter_size * variable
430 const size_t size_B = filter_total_size * dims.out_depth;
432 const size_t size_C = output_image_size * filter_total_size;
466 static_cast<int64>(filter_total_size)}),
500 TensorMap C(col_buffer_data, output_image_size, filter_total_size);
504 ConstTensorMap B(filter_data, filter_total_size, dims.out_depth);
532 &output_image_size, &filter_total_size,
542 MatrixMap C(im2col_buf, output_image_size, filter_total_size);
545 ConstMatrixMap B(filter_data, filter_total_size, dims.out_depth);
    [all...]
conv_grad_filter_ops.cc 390 const int filter_total_size = dims.spatial_dims[0].filter_size * variable
408 const size_t size_A = output_image_size * filter_total_size;
412 const size_t size_C = filter_total_size * dims.out_depth;
425 static_cast<int64>(filter_total_size)}),
447 TensorMap C(filter_backprop_data, filter_total_size, dims.out_depth);
483 filter_total_size);
    [all...]
deep_conv2d.cc 405 const int64 filter_total_size = local
416 filter_total_size + filter_transform_buffer_size + filter_out_buf_size;
    [all...]

Completed in 688 milliseconds