Home
last modified time | relevance | path

Searched refs:truncated_normal (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py53 return random_ops.truncated_normal(shape, seed=0, stddev=0.1)
88 x = random_ops.truncated_normal([2, 5, 5, 4], seed=0)
118 x1 = random_ops.truncated_normal([1, 784], seed=0)
119 x2 = random_ops.truncated_normal([1, 784], seed=0)
120 x3 = random_ops.truncated_normal([1, 784], seed=0)
121 x4 = random_ops.truncated_normal([1, 784], seed=0)
129 x1 = random_ops.truncated_normal([1, 784], seed=0)
130 x2 = random_ops.truncated_normal([1, 784], seed=0)
131 x3 = random_ops.truncated_normal([1, 784], seed=0)
132 x4 = random_ops.truncated_normal([1, 784], seed=0)
[all …]
Dcost_analyzer_test.py91 random_ops.truncated_normal([5, 5, 1, 32], stddev=0.1))
92 b = variables.Variable(random_ops.truncated_normal([32], stddev=0.1))
98 random_ops.truncated_normal([25088, 10], stddev=0.1))
99 b_fc = variables.Variable(random_ops.truncated_normal([10], stddev=0.1))
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dmnist_train.py61 tf.random.truncated_normal([5, 5, num_channels, n_hidden_1],
66 tf.random.truncated_normal([5, 5, n_hidden_1, n_hidden_2],
71 tf.random.truncated_normal([n_hidden_3, flatten_size],
76 tf.random.truncated_normal([num_classes, n_hidden_3],
/external/tensorflow/tensorflow/python/kernel_tests/random/
Drandom_ops_test.py166 rng = random_ops.truncated_normal(
236 n = random_ops.truncated_normal(v.shape)
243 rnd1 = random_ops.truncated_normal(shape, 0.0, 1.0, dtypes.float32)
244 rnd2 = random_ops.truncated_normal(shape, 0.0, 1.0, dtypes.float32)
422 rnd1 = random_ops.truncated_normal([1, 2, 3])
425 rnd2 = random_ops.truncated_normal(
429 rnd3 = random_ops.truncated_normal(array_ops.placeholder(dtypes.int32))
/external/tensorflow/tensorflow/python/ops/
Dinit_ops_v2.py488 return self._random_generator.truncated_normal(shape, self.mean,
594 return self._random_generator.truncated_normal(shape, 0.0, stddev, dtype)
1084 def truncated_normal(self, shape, mean, stddev, dtype): member in _RandomGenerator
1089 op = random_ops.truncated_normal
1101 truncated_normal = TruncatedNormal variable
Dinitializers_ns.py30 truncated_normal = init_ops.truncated_normal_initializer variable
Dconv2d_benchmark.py67 random_ops.truncated_normal(input_shape, dtype=dtype))
69 random_ops.truncated_normal(filter_shape, dtype=dtype))
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateless_random_ops.cc357 xla::XlaOp truncated_normal = TruncatedNormal(uniform); in Compile() local
358 truncated_normal = MaybeConvertF32ToBF16(truncated_normal, dtype_); in Compile()
359 ctx->SetOutput(0, truncated_normal); in Compile()
Dstateful_random_ops.cc321 xla::XlaOp truncated_normal = TruncatedNormal(uniform); in Compile() local
322 truncated_normal = MaybeConvertF32ToBF16(truncated_normal, dtype_); in Compile()
323 return {{truncated_normal, state}}; in Compile()
Dstateless_random_ops_v2.cc447 xla::XlaOp truncated_normal = TruncatedNormal(result.value); in Compile() local
448 truncated_normal = MaybeConvertF32ToBF16(truncated_normal, dtype_); in Compile()
449 ctx->SetOutput(0, truncated_normal); in Compile()
/external/tensorflow/tensorflow/python/keras/initializers/
Dinitializers_v2.py420 return self._random_generator.truncated_normal(shape, self.mean,
521 return self._random_generator.truncated_normal(shape, 0.0, stddev, dtype)
985 def truncated_normal(self, shape, mean, stddev, dtype): member in _RandomGenerator
990 op = random_ops.truncated_normal
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.initializers.-variance-scaling.pbtxt8 …d\'], varargs=None, keywords=None, defaults=[\'1.0\', \'fan_in\', \'truncated_normal\', \'None\'],…
Dtensorflow.initializers.truncated_normal.pbtxt1 path: "tensorflow.initializers.truncated_normal"
Dtensorflow.keras.initializers.truncated_normal.pbtxt1 path: "tensorflow.keras.initializers.truncated_normal"
Dtensorflow.initializers.variance_scaling.pbtxt8 …d\'], varargs=None, keywords=None, defaults=[\'1.0\', \'fan_in\', \'truncated_normal\', \'None\'],…
Dtensorflow.keras.initializers.variance_scaling.pbtxt8 …d\'], varargs=None, keywords=None, defaults=[\'1.0\', \'fan_in\', \'truncated_normal\', \'None\'],…
Dtensorflow.keras.initializers.-variance-scaling.pbtxt8 …d\'], varargs=None, keywords=None, defaults=[\'1.0\', \'fan_in\', \'truncated_normal\', \'None\'],…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.initializers.truncated_normal.pbtxt1 path: "tensorflow.initializers.truncated_normal"
Dtensorflow.variance_scaling_initializer.pbtxt8 …e\'], varargs=None, keywords=None, defaults=[\'1.0\', \'fan_in\', \'truncated_normal\', \'None\', …
Dtensorflow.keras.initializers.-variance-scaling.pbtxt8 …e\'], varargs=None, keywords=None, defaults=[\'1.0\', \'fan_in\', \'truncated_normal\', \'None\', …
Dtensorflow.initializers.variance_scaling.pbtxt8 …e\'], varargs=None, keywords=None, defaults=[\'1.0\', \'fan_in\', \'truncated_normal\', \'None\', …
Dtensorflow.keras.initializers.truncated_normal.pbtxt1 path: "tensorflow.keras.initializers.truncated_normal"
Dtensorflow.initializers.pbtxt36 name: "truncated_normal"
Dtensorflow.keras.initializers.pbtxt96 name: "truncated_normal"
/external/tensorflow/tensorflow/compiler/tests/
Drandom_ops_test.py117 return random_ops.truncated_normal(shape=[2], dtype=dtype)
173 x = random_ops.truncated_normal(shape=[count], dtype=dtype)

123