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

  /external/tensorflow/tensorflow/core/grappler/costs/
op_level_cost_estimator.cc 501 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { local
512 return conv_dims;
522 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( local
526 int64 ops = conv_dims.batch;
527 ops *= conv_dims.ox * conv_dims.oy;
528 ops *= conv_dims.kx * conv_dims.ky;
529 ops *= conv_dims.iz * conv_dims.oz
756 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( local
806 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( local
    [all...]
  /external/tensorflow/tensorflow/python/ops/
nn_ops.py 144 conv_dims = input_shape.ndims - 2
146 strides = [1] * conv_dims
147 elif len(strides) != conv_dims:
149 conv_dims))
150 if conv_dims == 1:
161 elif conv_dims == 2:
172 elif conv_dims == 3:
    [all...]

Completed in 461 milliseconds