Searched defs:ones_t (Results 1 – 2 of 2) sorted by relevance
983 Tensor ones_t(DT_FLOAT, TensorShape({64})); in TEST_F() local1019 const TensorProto& ones_t = node.attr().at("value").tensor(); in TEST_F() local
48 Tensor ones_t(DTYPE, TensorShape({2, 2})); in SimpleNeutralElementTest() local