Searched refs:ones_t (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | dense_update_ops_no_tsan_test.py | 37 ones_t = array_ops.fill([1024, 1024], 1.0)
41 p, ones_t, use_locking=False) for _ in range(20)
64 ones_t = array_ops.fill([1024, 1024], float(1))
67 state_ops.assign(p, math_ops.multiply(ones_t, float(i)), False)
97 ones_t = array_ops.fill([1024, 1024], 1.0)
101 p, ones_t, use_locking=True) for _ in range(20)
124 ones_t = array_ops.fill([1024, 1024], 1.0)
128 p, math_ops.multiply(ones_t, float(i)), use_locking=True)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 46 Tensor ones_t(DTYPE, TensorShape({2, 2})); in SimpleNeutralElementTest() local
50 ones_t.flat<T>()(i) = T(1); in SimpleNeutralElementTest()
54 Output ones = ops::Const(s.WithOpName("ones"), ones_t); in SimpleNeutralElementTest()
3756 Tensor ones_t(DT_FLOAT, TensorShape({64})); in TEST_F() local
3759 ones_t.flat<float>()(i) = 1.0f; in TEST_F()
3762 Output host_ones = ops::Const(scope.WithOpName("host_ones"), ones_t); in TEST_F()
3789 const TensorProto& ones_t = node.attr().at("value").tensor(); in TEST_F() local
3790 EXPECT_EQ(ones_t.float_val_size(), 1); in TEST_F()
3791 EXPECT_EQ(ones_t.float_val(0), 1.0f); in TEST_F()
|