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

  /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
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/
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/
cudnn_conv_padding_legalization.cc 245 auto* new_dim = new_backward_conv_window.mutable_dimensions(i); local
246 new_dim->set_padding_low(new_conv_padding);
247 new_dim->set_padding_high(new_conv_padding);
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/
types.h 286 const int new_dim = local
288 result.sizes[i] = new_dim;
290 cum_prod *= new_dim;
  /external/tensorflow/tensorflow/lite/kernels/internal/
types.h 302 const int new_dim = local
304 result.sizes[i] = new_dim;
306 cum_prod *= new_dim;
    [all...]
  /external/tensorflow/tensorflow/python/framework/
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])
    [all...]
  /external/tensorflow/tensorflow/core/framework/
shape_inference.h 452 // <new_dim>.
453 Status ReplaceDim(ShapeHandle s, int64 dim_index, DimensionHandle new_dim,
    [all...]
shape_inference.cc 712 DimensionHandle new_dim, ShapeHandle* out) {
727 dims[dim_index] = new_dim;
    [all...]
  /external/tensorflow/tensorflow/python/keras/layers/
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)
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_properties.cc 946 DimensionHandle new_dim = GetUnknownOutputDim(node, port_index, d); local
    [all...]
  /external/tensorflow/tensorflow/python/ops/parallel_for/
pfor.py     [all...]

Completed in 2262 milliseconds