Searched refs:s_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes.py | 134 def _valid(in_dim, k_dim, s_dim): argument
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)
146 def _same(in_dim, s_dim): argument
148 return (in_dim + s_dim - 1) // s_dim
152 output_size = [_same(in_dim, s_dim)
153 for in_dim, s_dim in zip(input_size, strides)]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 815 int s_dim = 1; in CountBatchMatMulOperations() local
817 s_dim = smaller_rank_shape->dim(s_i).size(); in CountBatchMatMulOperations()
819 num_matmuls *= std::max(b_dim, s_dim); in CountBatchMatMulOperations()
|