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

  /external/tensorflow/tensorflow/core/grappler/costs/
op_level_cost_estimator.cc 632 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { local
643 return conv_dims;
652 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( local
656 // in DepthwiseConv2dNative conv_dims.oz is actually the channel depth
658 // conv_dims.iz * conv_dims.oz. thus # ops = N x H x W x oz_effective x 2RS.
661 int64 ops = conv_dims.batch;
662 ops *= conv_dims.ox * conv_dims.oy;
663 ops *= conv_dims.kx * conv_dims.ky
951 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( local
1008 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( local
1479 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { local
    [all...]
  /external/tensorflow/tensorflow/contrib/layers/python/layers/
layers.py     [all...]
  /external/tensorflow/tensorflow/python/ops/
nn_ops.py 174 conv_dims = input_shape.ndims - 2
176 strides = [1] * conv_dims
177 elif len(strides) != conv_dims:
179 conv_dims))
180 if conv_dims == 1:
189 elif conv_dims == 2:
200 elif conv_dims == 3:
    [all...]

Completed in 2960 milliseconds