Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer_test.cc106 Output x = ops::RandomUniform(s.WithOpName("x"), {1, 2}, DT_FLOAT); in TEST_F()
107 Output y = ops::RandomUniform(s.WithOpName("y"), {1, 2}, DT_FLOAT); in TEST_F()
153 Output x = ops::RandomUniform(s.WithOpName("x"), {1, 2}, DT_FLOAT); in TEST_F()
242 Output x = ops::RandomUniform(s.WithOpName("x"), {1, 2}, DT_FLOAT); in TEST_F()
243 Output y = ops::RandomUniform(s.WithOpName("y"), {1, 2}, DT_FLOAT); in TEST_F()
264 Output x = ops::RandomUniform(s, {1, 2}, DT_FLOAT); in TEST_F()
295 Output x = ops::RandomUniform(s.WithOpName("x").WithDevice("/CPU:0"), {1, 2}, in TEST_F()
297 Output y = ops::RandomUniform(s.WithOpName("y").WithDevice("/CPU:0"), {1, 2}, in TEST_F()
330 Output x = ops::RandomUniform(s.WithOpName("x").WithDevice("/CPU:0"), {1, 2}, in TEST_F()
332 Output y = ops::RandomUniform(s.WithOpName("y").WithDevice("/CPU:0"), {1, 2}, in TEST_F()
[all …]
Dgeneric_layout_optimizer_transposer_test.cc106 ops::RandomUniform(scope->WithOpName("input"), in SimpleConv2D()
109 ops::RandomUniform(scope->WithOpName("filter"), in SimpleConv2D()
131 ops::RandomUniform(scope.WithOpName("x"), in CreateSimpleFusedBatchNorm()
134 ops::RandomUniform(scope.WithOpName("scale"), {kDepthIn}, DT_FLOAT); in CreateSimpleFusedBatchNorm()
136 ops::RandomUniform(scope.WithOpName("offset"), {kDepthIn}, DT_FLOAT); in CreateSimpleFusedBatchNorm()
138 ops::RandomUniform(scope.WithOpName("mean"), {kDepthIn}, DT_FLOAT); in CreateSimpleFusedBatchNorm()
139 auto var = ops::RandomUniform(scope.WithOpName("var"), {kDepthIn}, DT_FLOAT); in CreateSimpleFusedBatchNorm()
156 ops::RandomUniform(scope.WithOpName("orig_input"), in CreateSimpleMaxPoolGrad()
158 auto output_data = ops::RandomUniform( in CreateSimpleMaxPoolGrad()
162 ops::RandomUniform(scope.WithOpName("grad"), in CreateSimpleMaxPoolGrad()
[all …]
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_RandomUniform.pbtxt2 graph_op_name: "RandomUniform"
4 name: "random.RandomUniform"
/external/tensorflow/tensorflow/core/kernels/
Drandom_op_test.cc40 Graph* RandomUniform(int64_t n) { in RandomUniform() function
42 test::graph::RandomUniform(g, test::graph::Constant(g, VecShape(n)), in RandomUniform()
71 BM_RNG(cpu, RandomUniform);
75 BM_RNG(gpu, RandomUniform);
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.initializers.-random-uniform.pbtxt1 path: "tensorflow.initializers.RandomUniform"
3 is_instance: "<class \'keras.initializers.initializers_v2.RandomUniform\'>"
Dtensorflow.random_uniform_initializer.pbtxt3 is_instance: "<class \'tensorflow.python.ops.init_ops_v2.RandomUniform\'>"
Dtensorflow.initializers.random_uniform.pbtxt3 is_instance: "<class \'keras.initializers.initializers_v2.RandomUniform\'>"
/external/tensorflow/tensorflow/python/keras/initializers/
Dinitializers_v1.py62 class RandomUniform(init_ops.RandomUniform): class
66 super(RandomUniform, self).__init__(
D__init__.py61 LOCAL.ALL_OBJECTS['RandomUniformV2'] = initializers_v2.RandomUniform
104 'RandomUniform': initializers_v1.RandomUniform,
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc706 auto x = ops::RandomUniform( in CreateGrapplerItemWithConv2Ds()
708 auto y = ops::RandomUniform( in CreateGrapplerItemWithConv2Ds()
710 auto z = ops::RandomUniform( in CreateGrapplerItemWithConv2Ds()
712 auto f = ops::RandomUniform( in CreateGrapplerItemWithConv2Ds()
731 auto x = ops::RandomUniform( in CreateGrapplerItemWithConv2DAndVariable()
751 auto a = ops::RandomUniform(s.WithOpName("a"), {3200, 3200}, DT_FLOAT); in CreateGrapplerItemWithMatmulChain()
752 auto b = ops::RandomUniform(s.WithOpName("b").WithControlDependencies(a), in CreateGrapplerItemWithMatmulChain()
754 auto c = ops::RandomUniform(s.WithOpName("c").WithControlDependencies(b), in CreateGrapplerItemWithMatmulChain()
756 auto d = ops::RandomUniform(s.WithOpName("d").WithControlDependencies(c), in CreateGrapplerItemWithMatmulChain()
758 auto e = ops::RandomUniform(s.WithOpName("e").WithControlDependencies(d), in CreateGrapplerItemWithMatmulChain()
[all …]
Danalytical_cost_estimator_test.cc59 auto images = ops::RandomUniform( in CreateMiniGraph()
61 auto labels = ops::RandomUniform(s.WithOpName("label"), {batch, num_labels}, in CreateMiniGraph()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DRandomUniform.pbtxt2 name: "RandomUniform"
49 name: "RandomUniform"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DRandomUniform.pbtxt2 name: "RandomUniform"
49 name: "RandomUniform"
/external/tensorflow/tensorflow/python/ops/
Dinit_ops_v2_test.py198 init_ops_v2.RandomUniform(minval=-1, maxval=1, seed=124),
207 init1 = init_ops_v2.RandomUniform(0, 7, seed=1)
208 init2 = init_ops_v2.RandomUniform(0, 7, seed=1)
213 init1 = init_ops_v2.RandomUniform(0, 7, seed=1)
214 init2 = init_ops_v2.RandomUniform(0, 7, seed=2)
219 init = init_ops_v2.RandomUniform(0.0, 1.0)
224 init = init_ops_v2.RandomUniform(0, 7, seed=1)
Dinit_ops_v2.py298 class RandomUniform(Initializer): class
861 random_uniform_initializer = RandomUniform
1108 uniform = random_uniform = RandomUniform
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_RandomUniform.pbtxt2 graph_op_name: "RandomUniform"
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dlarge_function_graph.pbtxt28 op: "RandomUniform"
86 op: "RandomUniform"
144 op: "RandomUniform"
202 op: "RandomUniform"
/external/tensorflow/tensorflow/compiler/tests/
Dlstm_layer_inference.pbtxt73 name: "random_uniform/RandomUniform"
74 op: "RandomUniform"
118 input: "random_uniform/RandomUniform"
356 name: "inputs/random_uniform/RandomUniform"
357 op: "RandomUniform"
401 input: "inputs/random_uniform/RandomUniform"
583 name: "inputs/random_uniform_1/RandomUniform"
584 op: "RandomUniform"
628 input: "inputs/random_uniform_1/RandomUniform"
810 name: "inputs/random_uniform_2/RandomUniform"
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.random_uniform_initializer.pbtxt3 is_instance: "<class \'tensorflow.python.ops.init_ops.RandomUniform\'>"
Dtensorflow.initializers.random_uniform.pbtxt3 is_instance: "<class \'tensorflow.python.ops.init_ops.RandomUniform\'>"
/external/pytorch/test/onnx/expect/
DTestOperators.test_rand.expect8 op_type: "RandomUniform"
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmake_deterministic_test.cc360 {test::function::RandomUniform(), OuterRandomUniform()}); in TEST_P()
393 {test::function::RandomUniform(), OuterRandomUniform()}); in TEST_P()
436 {test::function::RandomUniform(), OuterRandomUniform()}); in TEST_P()
474 func = test::function::RandomUniform(); in TEST_P()
665 func = test::function::RandomUniform(); in TEST_P()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-prefer-tf2xla.mlir52 %0 = "tf.RandomUniform"(%arg0) : (tensor<3xi32>) -> tensor<12x?x64xf32>
67 …%0 = "tf.RandomUniform"(%cst) {seed = 87654321 : i64, seed2 = 0 : i64} : (tensor<4xi32>) -> tensor…
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_RandomUniform.pbtxt2 graph_op_name: "RandomUniform"
/external/tensorflow/tensorflow/core/framework/
Dfunction_testlib.h114 FunctionDef RandomUniform();

123