Searched defs:shape_vector (Results 1 – 2 of 2) sorted by relevance
379 std::vector<int64> shape_vector = {2, 2, 2, 2, 2, 2, 2, 2, 2}; in TEST_F() local
202 TensorShape shape_vector({input_channels}); in SimpleFusedBatchNormGrad() local