Searched refs:inp_dims (Results 1 – 1 of 1) sorted by relevance
176 std::vector<int64> inp_dims; in Compile() local177 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1 + i, &inp_dims)); in Compile()184 offset += inp_dims[j]; in Compile()187 const int32 inp_element = inp_dims[j]; in Compile()191 i, " has shape [", absl::StrJoin(inp_dims, " "), in Compile()