Searched refs:new_dim (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | core_layers.py | 220 new_dim = utils.conv_output_length(
226 new_space.append(new_dim)
233 new_dim = utils.conv_output_length(
239 new_space.append(new_dim)
|
| /external/tensorflow/tensorflow/contrib/kernel_methods/python/ |
| D | kernel_estimators.py | 93 new_dim = sum(mapper.output_dim for mapper in column_kernel_mappers)
95 layers.feature_column.real_valued_column(mapped_column_name, new_dim))
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_padding_legalization.cc | 245 auto* new_dim = new_backward_conv_window.mutable_dimensions(i); in CanonicalizeBackwardFilterConvolution() local
246 new_dim->set_padding_low(new_conv_padding); in CanonicalizeBackwardFilterConvolution()
247 new_dim->set_padding_high(new_conv_padding); in CanonicalizeBackwardFilterConvolution()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 286 const int new_dim = in ToRuntimeDims() local
288 result.sizes[i] = new_dim; in ToRuntimeDims()
290 cum_prod *= new_dim; in ToRuntimeDims()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 302 const int new_dim = in ToRuntimeDims() local
304 result.sizes[i] = new_dim; in ToRuntimeDims()
306 cum_prod *= new_dim; in ToRuntimeDims()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_util.py | 873 new_dim = tensor_shape.Dimension(None)
875 new_dim = tensor_shape.Dimension(pack_input_val)
876 ret = ret.concatenate([new_dim])
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional.py | 219 new_dim = conv_utils.conv_output_length(
225 new_space.append(new_dim)
232 new_dim = conv_utils.conv_output_length(
238 new_space.append(new_dim)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 712 DimensionHandle new_dim, ShapeHandle* out) { in ReplaceDim() argument
727 dims[dim_index] = new_dim; in ReplaceDim()
|
| D | shape_inference.h | 453 Status ReplaceDim(ShapeHandle s, int64 dim_index, DimensionHandle new_dim,
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 946 DimensionHandle new_dim = GetUnknownOutputDim(node, port_index, d); in OutputAsUnion() local
947 TF_CHECK_OK(ctx->ReplaceDim(relaxed, d, new_dim, &relaxed)); in OutputAsUnion()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1666 new_dim = array_ops.shape(t)[:1]
1667 new_shape = array_ops.concat([new_dim, shape], axis=0)
|