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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
conv_ops.cc 43 TensorShape expanded_shape = shape; local
44 expanded_shape.set_dim(num_dims - 1, shape.dim_size(num_dims - 2) *
46 return expanded_shape;
  /external/tensorflow/tensorflow/core/kernels/
serialize_sparse_op.cc 374 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank}));
376 auto expanded_shape_t = expanded_shape.vec<int64>();
380 TensorShape expanded_tensor_shape(expanded_shape.vec<int64>());
sparse_tensors_map_ops.cc 418 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank}));
421 auto expanded_shape_t = expanded_shape.vec<int64>();
  /external/tensorflow/tensorflow/python/ops/
nn_grad.py 315 expanded_shape = array_ops.concat([
321 expanded_shape = array_ops.concat(
325 expanded_grad = array_ops.reshape(received_grad, expanded_shape)
    [all...]
metrics_impl.py     [all...]
  /external/tensorflow/tensorflow/python/layers/
normalization.py 501 expanded_shape = [self.virtual_batch_size, -1] + original_shape[1:]
504 inputs = array_ops.reshape(inputs, expanded_shape)
    [all...]
  /external/tensorflow/tensorflow/python/ops/distributions/
distribution.py     [all...]

Completed in 357 milliseconds