Home
last modified time | relevance | path

Searched refs:in_dim (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py134 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)
146 def _same(in_dim, s_dim): argument
147 if in_dim is not None:
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/kernels/
Dmkl_avgpooling_op.cc75 mkl_context.params.in_dim = tensor_in.dims(); in Compute()
77 mkl_context.params.in_dim = mkl_context.input_shape.GetDimension(); in Compute()
128 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim, in Compute()
131 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); in Compute()
166 CHECK_EQ(dnnLayoutCreate_F32(&lt_user_input, params.in_dim, in MklCreateLayoutsAndPrimitives()
262 mkl_context.params.in_dim = out_backprop.dims(); in Compute()
264 mkl_context.params.in_dim = mkl_context.out_backprop_shape.GetDimension(); in Compute()
319 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim, in Compute()
322 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); in Compute()
381 CHECK_EQ(dnnLayoutCreate_F32(&lt_user_outbackprop, params.in_dim, in MklCreateLayoutsAndPrimitives()
[all …]
Dmkl_maxpooling_op.cc83 mkl_context.params.in_dim = 4; in Compute()
109 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim, in Compute()
112 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); in Compute()
158 CHECK_EQ(dnnLayoutCreate_F32(&lt_user_input, params.in_dim, in MklCreateLayoutsAndPrimitives()
247 mkl_context.params.in_dim = tensor_in.dims(); in Compute()
249 mkl_context.params.in_dim = mkl_context.input_shape.GetDimension(); in Compute()
282 mkl_output_shape.SetTfLayout(mkl_context.params.in_dim, in Compute()
285 mkl_output_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); in Compute()
324 CHECK_EQ(dnnLayoutCreate_F32(&lt_input_user, params.in_dim, in MklCreateLayouts()
334 CHECK_EQ(dnnLayoutCreate_F32(&lt_outbackprop_user, params.in_dim, in MklCreateLayouts()
Deigen_spatial_convolutions_test.cc1327 DSizes<ptrdiff_t, 4> in_dim; in TEST() local
1328 in_dim[0] = 1; in TEST()
1329 in_dim[1] = 3; in TEST()
1330 in_dim[2] = 4; in TEST()
1331 in_dim[3] = 1; in TEST()
1350 .reshape(in_dim); in TEST()
1361 .reshape(in_dim); in TEST()
Dmkl_pooling_ops_common.h759 size_t in_dim; member
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc1211 const int64 in_dim = out_dim + offset; in ComputeFlatInnerDims() local
1212 out_dims[out_dim] = in_dim < 0 ? 1 : orig[in_dim]; in ComputeFlatInnerDims()
1214 for (int64 in_dim = 0; in_dim < offset; ++in_dim) { in ComputeFlatInnerDims() local
1215 out_dims[0] *= orig[in_dim]; in ComputeFlatInnerDims()
1226 for (int64 in_dim = num_out_dims; in_dim < orig.size(); ++in_dim) { in ComputeFlatOuterDims() local
1227 out_dims[num_out_dims - 1] *= orig[in_dim]; in ComputeFlatOuterDims()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc490 auto in_dim = c->Dim(in, in_rank - i - 1); in __anon7c94107b0702() local
491 if (c->Value(in_dim) > 1) { in __anon7c94107b0702()
495 TF_RETURN_IF_ERROR(c->Merge(in_dim, out_dim, &out_dim)); in __anon7c94107b0702()