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

  /external/tensorflow/tensorflow/core/kernels/
parameterized_truncated_normal_op.cc 256 const Tensor& stddevs_tensor = ctx->input(2); variable
286 OP_REQUIRES(ctx, stddevs_tensor.dims() <= 1,
289 stddevs_tensor.shape().DebugString()));
300 (stddevs_tensor.dims() == 0 || stddevs_tensor.dim_size(0) == 1) &&
323 TensorShapeUtils::IsScalar(stddevs_tensor.shape()) ||
324 stddevs_tensor.dim_size(0) == 1 ||
325 stddevs_tensor.dim_size(0) == num_batches,
328 stddevs_tensor.shape().DebugString()));
354 stddevs_tensor.flat<T>(), minvals_tensor.flat<T>()
    [all...]
  /external/tensorflow/tensorflow/python/ops/
random_ops.py 125 stddevs_tensor = ops.convert_to_tensor(stddevs, dtype=dtype, name="stddevs")
132 stddevs_tensor,

Completed in 199 milliseconds