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

  /external/tensorflow/tensorflow/core/grappler/costs/
op_level_cost_estimator.cc 562 double m_dim, n_dim, k_dim, k_dim_b = 0; local
580 k_dim = a_matrix_shape.dim(0).size();
583 k_dim = a_matrix_shape.dim(1).size();
593 VLOG(1) << "M, N, K: " << m_dim << "," << n_dim << "," << k_dim; local
596 if (k_dim_b != 1 && k_dim != 1 && k_dim_b != k_dim) {
600 // One of k_dim and k_dim_b might be 1 (mininum dimension size).
601 k_dim = std::max(k_dim, k_dim_b);
604 ops = m_dim * n_dim * k_dim * 2
    [all...]
  /external/tensorflow/tensorflow/python/framework/
common_shapes.py 122 def _valid(in_dim, k_dim, s_dim):
123 if in_dim is not None and k_dim is not None:
124 return (in_dim - k_dim + s_dim) // s_dim
129 _valid(in_dim, k_dim, s_dim)
130 for in_dim, k_dim, s_dim in zip(input_size, filter_size, strides)
  /external/tensorflow/tensorflow/core/ops/
nn_ops.cc 1143 DimensionHandle k_dim; local
    [all...]

Completed in 340 milliseconds