Home
last modified time | relevance | path

Searched refs:fan_out (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/data/benchmarks/
Dmap_benchmark.py57 def benchmark_helper(fan_out, fn, use_inter_op_parallelism, label): argument
59 tuple(0 for _ in range(fan_out))).repeat(None)
65 name="fan_out_%d%s" % (fan_out, label))
67 for fan_out in fan_outs:
68 benchmark_helper(fan_out, lambda *xs: [x + 1 for x in xs], True, "")
69 benchmark_helper(fan_out, lambda *xs: [x + 1 for x in xs], False,
71 benchmark_helper(fan_out, lambda *xs: xs, True, "_short_circuit")
/external/tensorflow/tensorflow/python/keras/
Dinitializers_test.py44 fan_in = fan_out = 1
46 fan_in = fan_out = shape[0]
49 fan_out = shape[1]
57 fan_out = shape[-1] * receptive_field_size
58 return int(fan_in), int(fan_out)
128 fan_in, fan_out = _compute_fans(tensor_shape)
129 std = np.sqrt(2. / (fan_in + fan_out))
161 fan_in, fan_out = _compute_fans(tensor_shape)
162 std = np.sqrt(2. / (fan_in + fan_out))
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree_test.cc540 const int fan_out = state.range(1); in BM_Construct() local
544 std::vector<xla::Shape> shapes(fan_out, shape); in BM_Construct()
555 const int fan_out = state.range(1); in BM_ConstructUnowned() local
559 std::vector<xla::Shape> shapes(fan_out, shape); in BM_ConstructUnowned()
570 const int fan_out = state.range(1); in BM_Copy() local
574 std::vector<xla::Shape> shapes(fan_out, shape); in BM_Copy()
587 const int fan_out = state.range(1); in BM_Move() local
591 std::vector<xla::Shape> shapes(fan_out, shape); in BM_Move()
604 const int fan_out = state.range(1); in BM_ForEach() local
608 std::vector<xla::Shape> shapes(fan_out, shape); in BM_ForEach()
[all …]
/external/tensorflow/tensorflow/python/keras/initializers/
Dinitializers_v2.py509 fan_in, fan_out = _compute_fans(shape)
515 scale /= max(1., fan_out)
517 scale /= max(1., (fan_in + fan_out) / 2.)
1005 fan_in = fan_out = 1
1007 fan_in = fan_out = shape[0]
1010 fan_out = shape[1]
1018 fan_out = shape[-1] * receptive_field_size
1019 return int(fan_in), int(fan_out)
/external/tensorflow/tensorflow/python/ops/
Dinit_ops_test.py116 fan_in, fan_out = init_ops._compute_fans(tensor_shape)
117 std = np.sqrt(2. / (fan_in + fan_out))
152 fan_in, fan_out = init_ops._compute_fans(tensor_shape)
153 std = np.sqrt(2. / (fan_in + fan_out))
Dinit_ops.py501 fan_in, fan_out = _compute_fans(scale_shape)
505 scale /= max(1., fan_out)
507 scale /= max(1., (fan_in + fan_out) / 2.)
1411 fan_in = fan_out = 1
1413 fan_in = fan_out = shape[0]
1416 fan_out = shape[1]
1424 fan_out = shape[-1] * receptive_field_size
1425 return int(fan_in), int(fan_out)
Dinit_ops_v2.py582 fan_in, fan_out = _compute_fans(shape)
588 scale /= max(1., fan_out)
590 scale /= max(1., (fan_in + fan_out) / 2.)
Dinit_ops_v2_test.py532 fan_in, fan_out = init_ops_v2._compute_fans(shape)
533 std = np.sqrt(2. / (fan_in + fan_out))
543 fan_in, fan_out = init_ops_v2._compute_fans(shape)
544 std = np.sqrt(2. / (fan_in + fan_out))
/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer_test.cc178 const int fan_out = state.range(1); in BM_TakeSubTree() local
183 std::vector<xla::Shape> shapes(fan_out, shape); in BM_TakeSubTree()
191 (void)shaped_buffer.TakeSubTree(/*index=*/{fan_out / 2}).release(); in BM_TakeSubTree()
/external/igt-gpu-tools/tests/i915/
Dgem_stress.c422 static void fan_out(void) in fan_out() function
876 fan_out();
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc5326 int64 fan_out = 1; in ComputeReductionCodegenInfo() local
5328 fan_out = fusion.getFusionResults().size(); in ComputeReductionCodegenInfo()
5333 int64 max_block_size = std::max(64LL, 512LL / NearestPowerOfTwo(fan_out)); in ComputeReductionCodegenInfo()