Searched defs:axes_shape (Results 1 – 2 of 2) sorted by relevance
76 const TensorShape axes_shape = ctx->InputShape(1); in Compile() local
118 const std::array<int32_t, 1> axes_shape{ in CreateTfLiteModel() local