| /external/tensorflow/tensorflow/core/kernels/ |
| D | parameterized_truncated_normal_op_gpu.cu.cc | 57 const T* __restrict__ stddevs, bool single_stddev, in TruncatedNormalKernel() argument
95 const T input_stddev = stddevs[single_stddev ? 0 : batch_id]; in TruncatedNormalKernel()
239 typename TTypes<T>::ConstFlat stddevs, in operator ()()
249 num_elements, means.data(), means.dimension(0) == 1, stddevs.data(), in operator ()()
250 stddevs.dimension(0) == 1, minvals.data(), minvals.dimension(0) == 1, in operator ()()
|
| D | parameterized_truncated_normal_op.h | 41 typename TTypes<T>::ConstFlat stddevs,
56 typename TTypes<T>::ConstFlat stddevs,
|
| D | parameterized_truncated_normal_op.cc | 56 typename TTypes<T>::ConstFlat stddevs, in operator ()()
71 auto do_work = [samples_per_batch, num_elements, &ctx, &means, &stddevs, in operator ()()
99 T stddev = stddevs((stddevs.dimension(0) == 1) ? 0 : b); in operator ()()
319 typename TTypes<T>::ConstFlat stddevs, in operator ()()
335 &stddevs, &minvals, &maxvals, &gen, &output, in operator ()()
382 stddev = stddevs(stddevs_batch_indices[batch_idx]); in operator ()()
387 stddev = stddevs(batch_idx); in operator ()()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_StatelessParameterizedTruncatedNormal.pbtxt | 23 name: "stddevs"
51 `shape`, `seed`, `minvals`, `maxvals`, `means` and `stddevs`.
|
| /external/walt/android/WALT/app/src/main/java/org/chromium/latency/walt/ |
| D | Utils.java | 159 double[] stddevs = new double[steps]; in findBestShift() local
163 …stddevs[i] = getShiftError(laserT, touchT, touchY, bestShift + shiftStep * i - shiftStep * steps /… in findBestShift()
165 bestShift = argmin(stddevs) * shiftStep + bestShift - shiftStep * steps / 2; in findBestShift()
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | parameterized_truncated_normal_op_test.py | 287 shape=(int(1e5),), means=0.8, stddevs=0.05, minvals=-1., maxvals=1.)
297 stddevs=0.05,
446 stddevs=1.0,
453 stddevs=1.0,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateless_random_ops.py | 677 stddevs=1.0, argument
729 [shape, means, stddevs, minvals, maxvals]) as name:
732 stddevs_tensor = ops.convert_to_tensor(stddevs, name="stddevs")
|
| D | random_ops.py | 107 stddevs=1.0, argument
140 [shape, means, stddevs, minvals, maxvals]) as name:
143 stddevs_tensor = ops.convert_to_tensor(stddevs, dtype=dtype, name="stddevs")
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | StatelessParameterizedTruncatedNormal.pbtxt | 16 name: "stddevs"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 344 TF_ASSIGN_OR_RETURN(xla::XlaOp stddevs, in Compile()
350 return ParameterizedTruncatedNormal(uniform, means, stddevs, minvals, in Compile()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | random_ops_test.py | 187 stddevs=sigma,
206 stddevs=sigma,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.random.pbtxt | 85 …argspec: "args=[\'shape\', \'seed\', \'means\', \'stddevs\', \'minvals\', \'maxvals\', \'name\'], …
|
| D | tensorflow.raw_ops.pbtxt | 4593 …argspec: "args=[\'shape\', \'seed\', \'means\', \'stddevs\', \'minvals\', \'maxvals\', \'name\'], …
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.random.pbtxt | 97 …argspec: "args=[\'shape\', \'seed\', \'means\', \'stddevs\', \'minvals\', \'maxvals\', \'name\'], …
|
| D | tensorflow.raw_ops.pbtxt | 4593 …argspec: "args=[\'shape\', \'seed\', \'means\', \'stddevs\', \'minvals\', \'maxvals\', \'name\'], …
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 14775 …F_Float64]>, [{The standard deviation parameter of each batch. Must be greater than 0.}]>:$stddevs,
14783 `shape`, `seed`, `minvals`, `maxvals`, `means` and `stddevs`.}]>:$output
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 51702 name: "stddevs"
|