Searched refs:fill_shape_tensor (Results 1 – 1 of 1) sorted by relevance
487 Tensor fill_shape_tensor(DT_INT32, TensorShape({4})); in TEST() local488 fill_shape_tensor.vec<int32>()(0) = 1; in TEST()489 fill_shape_tensor.vec<int32>()(1) = 256; in TEST()490 fill_shape_tensor.vec<int32>()(2) = 1024; in TEST()491 fill_shape_tensor.vec<int32>()(3) = 1024; in TEST()492 Node* fill_shape_node = test::graph::Constant(&graph, fill_shape_tensor); in TEST()