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

  /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)
134 def _same(in_dim, s_dim):
135 if in_dim is not None:
136 return (in_dim + s_dim - 1) // s_dim
140 output_size = [_same(in_dim, s_dim)
141 for in_dim, s_dim in zip(input_size, strides)
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
mkl_avgpooling_op.cc 75 mkl_context.params.in_dim = tensor_in.dims();
77 mkl_context.params.in_dim = mkl_context.input_shape.GetDimension();
128 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim,
131 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_);
166 CHECK_EQ(dnnLayoutCreate_F32(&lt_user_input, params.in_dim,
262 mkl_context.params.in_dim = out_backprop.dims();
264 mkl_context.params.in_dim = mkl_context.out_backprop_shape.GetDimension();
319 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim,
322 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_);
381 CHECK_EQ(dnnLayoutCreate_F32(&lt_user_outbackprop, params.in_dim,
    [all...]
mkl_maxpooling_op.cc 82 mkl_context.params.in_dim = 4;
108 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim,
111 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_);
157 CHECK_EQ(dnnLayoutCreate_F32(&lt_user_input, params.in_dim,
245 mkl_context.params.in_dim = tensor_in.dims();
247 mkl_context.params.in_dim = mkl_context.input_shape.GetDimension();
280 mkl_output_shape.SetTfLayout(mkl_context.params.in_dim,
283 mkl_output_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_);
322 CHECK_EQ(dnnLayoutCreate_F32(&lt_input_user, params.in_dim,
332 CHECK_EQ(dnnLayoutCreate_F32(&lt_outbackprop_user, params.in_dim,
    [all...]
eigen_spatial_convolutions_test.cc 1325 DSizes<ptrdiff_t, 4> in_dim; local
    [all...]
mkl_fused_batch_norm_op.cc 134 mkl_shape_output.SetTfLayout(mkl_context.mkl_params.in_dim,
137 mkl_shape_output.SetTfDimOrder(mkl_context.mkl_params.in_dim,
219 size_t in_dim; member in struct:tensorflow::MklFusedBatchNormOp::__anon39842::MklBatchNormParams
244 mkl_params.in_dim = input_in_mkl_format
272 dnnLayoutCreate_F32(&mkl_lt_input, mkl_params.in_dim,
    [all...]
mkl_pooling_ops_common.h 404 size_t in_dim; member in struct:tensorflow::__anon39849
  /external/tensorflow/tensorflow/core/framework/
tensor.cc 1052 const int64 in_dim = out_dim + offset; local
    [all...]

Completed in 428 milliseconds