Searched refs:k_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes.py | 134 def _valid(in_dim, k_dim, s_dim): argument
135 if in_dim is not None and k_dim is not None:
136 return (in_dim - k_dim + s_dim) // s_dim
141 _valid(in_dim, k_dim, s_dim)
142 for in_dim, k_dim, s_dim in zip(input_size, filter_size, strides)
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 704 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local
722 k_dim = a_matrix_shape.dim(0).size(); in CountMatMulOperations()
725 k_dim = a_matrix_shape.dim(1).size(); in CountMatMulOperations()
735 VLOG(1) << "M, N, K: " << m_dim << "," << n_dim << "," << k_dim; in CountMatMulOperations()
738 if (k_dim_b != 1 && k_dim != 1 && k_dim_b != k_dim) { in CountMatMulOperations()
743 k_dim = std::max(k_dim, k_dim_b); in CountMatMulOperations()
746 ops = m_dim * n_dim * k_dim * 2; in CountMatMulOperations()
752 mat_mul->k = k_dim; in CountMatMulOperations()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 1204 DimensionHandle k_dim; in TopKShapeFn() local
1206 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &k_dim)); in TopKShapeFn()
1213 k_dim = c->MakeDim(k); in TopKShapeFn()
1217 if (c->ValueKnown(last_dim) && c->ValueKnown(k_dim) && in TopKShapeFn()
1218 c->Value(last_dim) < c->Value(k_dim)) { in TopKShapeFn()
1220 "input must have last dimension >= k = ", c->Value(k_dim), " but is ", in TopKShapeFn()
1227 TF_RETURN_IF_ERROR(c->Concatenate(s, c->Vector(k_dim), &s)); in TopKShapeFn()
|