Searched defs:inp_shape (Results 1 – 3 of 3) sorted by relevance
32 int* inp_shape, in prepare_data()100 int inp_shape[kNnlibMaxDim]; in mean_dim_out() local
173 const TensorShape inp_shape = ctx->InputShape(1 + i); in Compile() local
243 def triton_relu_flops(inp_shape, *args, **kwargs): argument