Home
last modified time | relevance | path

Searched refs:ones_t (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Ddense_update_ops_no_tsan_test.py41 ones_t = array_ops.fill([1024, 1024], 1.0)
45 p, ones_t, use_locking=False) for _ in range(20)
71 ones_t = array_ops.fill([1024, 1024], float(1))
74 state_ops.assign(p, math_ops.multiply(ones_t, float(i)), False)
107 ones_t = array_ops.fill([1024, 1024], 1.0)
111 p, ones_t, use_locking=True) for _ in range(20)
137 ones_t = array_ops.fill([1024, 1024], 1.0)
141 p, math_ops.multiply(ones_t, float(i)), use_locking=True)
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc968 Tensor ones_t(DT_FLOAT, TensorShape({64})); in TEST_F() local
971 ones_t.flat<float>()(i) = 1.0f; in TEST_F()
974 Output host_ones = ops::Const(scope.WithOpName("host_ones"), ones_t); in TEST_F()
1005 const TensorProto& ones_t = node.attr().at("value").tensor(); in TEST_F() local
1006 EXPECT_EQ(ones_t.float_val_size(), 1); in TEST_F()
1007 EXPECT_EQ(ones_t.float_val(0), 1.0f); in TEST_F()
Dconstant_folding_test.cc48 Tensor ones_t(DTYPE, TensorShape({2, 2})); in SimpleNeutralElementTest() local
52 ones_t.flat<T>()(i) = T(1); in SimpleNeutralElementTest()
56 Output ones = ops::Const(s.WithOpName("ones"), ones_t); in SimpleNeutralElementTest()