Searched defs:tensor_out_shape (Results 1 – 2 of 2) sorted by relevance
355 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile() local577 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile() local
222 const TensorShape& tensor_out_shape, in Compute()